Staff Machine Learning Engineer, Infrastructure
Waymo · Simulation (7XW)
- Location
- Mountain View, CA, USA; San Francisco, CA, USA
- Metro
- San Francisco Bay Area
- Workplace
- Unspecified
- Level
- Staff+
- Type
- Unspecified
- Pay
- $251K–$310K/yr (estimated from posting text)
- Posted
- 11 months ago
- Source
- Greenhouse
Read the full posting & apply ↗
What the posting covers
About the company
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…
What You'll Do
Design, build, and optimize realistic simulation environments and business logic running on TPUs using JAX and TensorFlow. Implement and optimize large-scale model and data parallelism strategies for training and running foundation models on TPU hardware.…
Minimum Qualifications
6+ years of professional software engineering experience, with at least 4 years focused on machine learning infrastructure (scaling, training, optimizing, and deploying large-scale ML systems). Direct ML programming experience on TPU and GPU hardware using…
Preferred Qualifications
Practical experience in Reinforcement Learning (RL), Sim2Real transfer, or Robotics. Experience with distributed ML frameworks and accelerators like GPU/TPU. Domain familiarity with Autonomous Driving systems, multi-agent simulations, or realistic world…
Excerpts only. The employer’s page has the complete posting.