Skip to content
← Waymo

Staff Machine Learning Engineer, AI Infrastructure

Waymo · Mountain View, CA, USA; San Francisco, CA, USA; New York, NY, USA · Mountain View (US-MTV-EMF680) · New York City (US-NYC-CHEL) · San Francisco (US-SFO-MKT555)

RoboticsonsiteStaff$251k–$310k
Apply on the lab’s site ↗

At a glance

Highlights

  • Lead ML infrastructure for evaluating the world's most experienced autonomous driver.
  • Generate rare driving events and scale simulation across billions of virtual miles.
  • Power metrics that evaluate behavior across 10M+ autonomous rider trips.

What they offer

  • Annual bonus program
  • Equity incentive plan
  • Generous Company benefits program

Ideal candidate

A seasoned ML systems builder who thrives on ambiguous, large-scale problems and mentoring others. Passionate about autonomous driving safety and turning evaluation insights into reliable production systems.

Requirements

  • B.S. in Computer Science, Machine Learning, or related technical field
  • 7+ years hands-on experience building machine learning systems
  • Experience designing, scaling, training and deploying large-scale ML systems
  • Strong coding experience in Python and/or C++
  • Experience with ML frameworks such as JAX, TensorFlow, or PyTorch
  • Proven ability to set technical direction in ambiguous problem spaces

Lab profile

About Waymo

Autonomous driving technology company building the Waymo Driver for fully autonomous ride-hailing.

What they build

  • Waymo Driver
  • Fully autonomous ride-hail service

Known for

  • Large-scale driving simulation
  • Simulation realism
  • Quantitative evaluations
  • Onboard autonomy models

Key facts

  • Founded in 2009
  • Started as the Google Self-Driving Car Project
More about Waymo →
Apply on the lab’s site ↗

Posted · Last seen