22 days ago
Base Salary
$213k - $263k/yr
Responsibilities
- Build scalable systems for training and fine-tuning large-scale models that evaluate driving behaviors.
- Develop and operate machine learning and data systems, simulation workflows, and evaluation tools.
- Design and scale distributed systems for dataset generation, model training, and evaluation across the ML lifecycle.
- Contribute to production, deployment, and optimization of machine learning models for Waymo’s vehicle fleet.
- Provide architectural guidance, own complex systems, and drive technical decisions.
- Collaborate cross-functionally to define system requirements and translate business goals into measurable technical deliverables.
Requirements
- M.S. or Ph.D. in Computer Science, Machine Learning, Artificial Intelligence, or a related technical field, or equivalent practical experience.
- At least 5 years of experience developing, designing, scaling, training, deploying, and optimizing large-scale machine learning systems and infrastructure.
- Contributions to machine learning tooling and frameworks such as PyTorch, Jax, Tensorflow, Ray, or similar, with understanding of user-facing APIs and internal workings.
- Strong expertise in distributed training, including gradient sharding and optimization strategies for scaling large models.
- Experience with ML accelerator profiling tools to identify performance bottlenecks.
- Preferred: 7 or more years of machine learning infrastructure experience.
- Preferred: experience in autonomous vehicles, robotics, or complex simulation environments.
- Preferred: familiarity with large-scale simulation platforms and their integration with ML training workflows.
Benefits
- Full-time position available across U.S. locations.
- Eligibility for a discretionary annual bonus program, equity incentive plan, and company benefits program, subject to eligibility requirements.
Tech Stack
PyTorchTensorFlow
Categories
Data EngineeringML Engineering
About Waymo
Waymo develops the Waymo Driver, a full-stack autonomous driving system, and operates the Waymo One robotaxi service that offers paid, driverless rides in select U.S. cities. The company monetizes through ride-hailing fares and partnerships integrating its technology on multiple vehicle platforms. Founded in 2009 as Google’s self-driving project, Waymo is headquartered in Mountain View and is an Alphabet subsidiary, with over 10 million rider-only trips and 100+ million autonomous miles on public roads.