Software Engineer, Machine Learning Performance

Reposted 9 Days Ago
Be an Early Applicant
2 Locations
In-Office
3M-3M Annually
Mid level
Automotive
The Role
Design and implement technical solutions to optimize machine learning models' performance by analyzing data, collaborating with teams, and contributing to engineering quality.
Summary Generated by Built In

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver powers Waymo’s fully autonomous ride-hail service and can also be applied to a range of vehicle platforms and product use cases. The Waymo Driver has provided over ten million rider-only trips, enabled by its experience autonomously driving over 100 million miles on public roads and tens of billions in simulation across 15+ U.S. states.

Waymo's Compute Team is tasked with a critical and exciting mission: We deliver the compute platform responsible for running the fully autonomous vehicle’s software stack. To achieve our mission, we architect and create high-performance custom silicon; we develop system-level compute architectures that push the boundaries of performance, power, and latency; and we collaborate closely with many other teammates to ensure we design and optimize hardware and software for maximum performance. We are a multidisciplinary team seeking curious and talented teammates to work on one of the world’s highest performance automotive compute platforms.

This role follows a hybrid work schedule, and you will report to the Tech Lead Manager of the Machine Learning Performance team.


You will:

  • Design and implement technical solutions for performance optimization of individual ML models
  • Execute end-to-end tasks consistently and in a timely manner, contributing directly to larger performance goals for ML accelerators
  • Collect, trace, and analyze application/ML model performance data to identify specific optimization opportunities, and prototype solutions at the application or compiler
  • Collaborate with infrastructure teams (e.g., compiler, runtime) to implement and land performance-critical optimizations. 
  • Uphold engineering quality by contributing to design documents, participating in timely code reviews, and writing comprehensive unit tests for your work products
  • Proactively spot and suggest areas of future performance work within the team's scope

You have:

  • BS degree in Computer Science/Electrical Engineering or equivalent, or equivalent practical experience
  • 3+ years of experience writing complex C++ code
  • 3+ years of experience writing code in Python
  • 1+ years experience in optimizing compute performance for ML applications
  • Experience in compute architectures and performance analysis optimization methodologies

We prefer:

  • Experience in ML compiler design and implementation
  • Experience in performance tools, simulators, HW/SW codesign
  • Experience in robotic application development/optimizations
  • Proficiency in collaboration with application teams and infrastructure teams

The expected base salary range for this full-time position is listed below. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level.  Waymo employees are also eligible to participate in Waymo’s discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements.

Salary Range
$2,600,000$3,150,000 TWD

Skills Required

  • BS degree in Computer Science/Electrical Engineering or equivalent
  • 3+ years of experience writing complex C++ code
  • 3+ years of experience writing code in Python
  • 1+ years experience optimizing compute performance for ML applications
  • Experience in compute architectures and performance analysis optimization methodologies
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
Mountain View, CA
2,359 Employees
Year Founded: 2009

What We Do

Waymo is an autonomous driving technology company with a mission to make it safe and easy for people and things to move around. With the Waymo Driver, we can improve the world’s mobility while saving thousands of lives. Waymo reaches out to candidates from official channels only (e.g. directly from @waymo.com email addresses, or through our recruiters or sourcers who are noted as such on LinkedIn). We do not contact candidates about career opportunities through instant messaging apps like Telegram, email addresses from domains other than waymo.com (such as Gmail addresses), direct messages on Twitter, Facebook, and Instagram, or text messages. Visit waymo.com to check out our official job listings.

Similar Jobs

In-Office
2 Locations
2359 Employees
4M-4M Annually

Micron Technology Logo Micron Technology

Operations Specialist

Artificial Intelligence • Hardware • Information Technology • Machine Learning
In-Office
Taipei City, TWN
45000 Employees

Ericsson Logo Ericsson

Integration Engineer

Cloud • Information Technology • Internet of Things • Machine Learning • Software • Cybersecurity • Infrastructure as a Service (IaaS)
In-Office
Taipei City, TWN
88000 Employees

Ericsson Logo Ericsson

Support Engineer

Cloud • Information Technology • Internet of Things • Machine Learning • Software • Cybersecurity • Infrastructure as a Service (IaaS)
In-Office
Taipei City, TWN
88000 Employees

Similar Companies Hiring

Cox Enterprises Thumbnail
Artificial Intelligence • Automotive • Greentech • Information Technology • Machine Learning • Software • Cybersecurity
Atlanta, GA
50000 Employees
UL Solutions Thumbnail
Automotive • Professional Services • Software • Consulting • Energy • Chemical • Renewable Energy
Chicago, IL
15000 Employees
HERE Technologies Thumbnail
Artificial Intelligence • Automotive • Computer Vision • Information Technology • Internet of Things • Logistics • Software
Amsterdam, NL
6000 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account