Machine Learning Engineer (Simulation Realism)

Location: Hybrid role in Mountain View, California, United States

Salary: $213,000–$263,000 USD per year

Company

Waymo develops autonomous driving technology and fully autonomous ride-hailing services through the Waymo Driver.

What you will do

• Develop ultra-realistic simulation technologies for autonomous vehicles and embodied agents.
• Apply machine learning and generative AI to improve simulation realism and steerability.
• Model realistic vehicles, pedestrians, cyclists, roads, traffic control systems, and weather conditions.
• Develop foundation-model-based world modeling, reasoning, and reinforcement learning solutions.
• Collaborate with foundational AI research teams to transfer research into scalable, production-ready systems.

Requirements

• 5+ years of experience in machine learning software engineering.
• 5+ years of Python experience.
• 5+ years of experience with machine learning frameworks including PyTorch, TensorFlow, or JAX/Flax.
• Experience developing ML models for robotics or self-driving vehicles is preferred.
• Strong production C++ experience and experience integrating ML models into production systems are preferred.
• Experience training large-scale models on GPU or TPU clusters is preferred.

Culture & Benefits

• Work with a diverse and collaborative group of software engineers, ML engineers, and data scientists.
• Participate in Waymo's discretionary annual bonus program and equity incentive plan.
• Access a company benefits program, subject to eligibility requirements.