Senior Machine Learning Engineer (Infra), Driver Understanding and Evaluation

Highlights

• Build scalable ML systems to evaluate interesting driving behaviors.
• Power evaluation for a fleet driving millions of miles autonomously.
• Improve evaluation and developer journeys for the Waymo Driver.

What they offer

• Discretionary annual bonus program
• Equity incentive plan
• Generous Company benefits program

Ideal candidate

An ML infrastructure builder with 5+ years scaling models who is passionate about autonomous vehicle evaluation and thrives at the intersection of data, simulation, and owning complex systems end-to-end.

Requirements

• M.S. or Ph.D. in Computer Science, Machine Learning, AI or related field or equivalent experience
• 5+ years in ML infrastructure developing and scaling large-scale ML systems
• History of contributions to ML tooling and frameworks e.g. PyTorch, JAX, TensorFlow, Ray
• Strong expertise in distributed training, gradient sharding and scaling large models