Senior Machine Learning Engineer (Infra)

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.