Software Engineer, Machine Learning Performance

Reposted 11 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

Waymo Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Waymo and has not been reviewed or approved by Waymo.

  • Fair & Transparent Compensation Pay is considered strong for full‑time technical roles, aligning with top‑tier AV/Big Tech benchmarks. Employer materials also emphasize competitive pay with eligibility for bonuses and equity as part of total rewards.
  • Healthcare Strength Coverage spans medical, dental, vision, mental‑health resources, onsite wellness centers, and counseling, with specialized programs such as menopause and transgender medical advocacy. Dependents are included, indicating depth across core health needs.
  • Parental & Family Support Offerings include fertility and family‑forming assistance, parental and baby‑bonding leave, caregiver and elder‑care support, and backup childcare. These programs are positioned to support families through multiple life stages.

Waymo Insights

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
HQ: 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

Ericsson Logo Ericsson

Senior Commercial Manager

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

Zscaler Logo Zscaler

Account Executive

Cloud • Information Technology • Security • Software • Cybersecurity
Easy Apply
Remote or Hybrid
Taiwan
8697 Employees

Snap Inc. Logo Snap Inc.

Manager, Product Design Engineering

Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Remote or Hybrid
Taipei City, TWN
5000 Employees
11-11 Annually

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