Machine Learning Engineer (Simulation Realism)

Location: Hybrid / Onsite in Mountain View, California

Salary: $204,000—$259,000 USD

Company

Autonomous driving technology company building the Waymo Driver for fully autonomous ride-hail services and other vehicle platforms.

What you will do

• Drive innovation in simulation technologies using foundation models for realistic embodied agents like vehicles, pedestrians, and cyclists.
• Apply ML expertise in Generative AI to enhance simulation realism for roads, traffic, and weather.
• Collaborate with AI research team to productionize world understanding, reasoning, and reinforcement learning solutions.
• Build infrastructure for counterfactual scenarios to test autonomous driving performance.

Requirements

• 5+ years of experience with ML software engineering
• 5+ years of experience with Python
• 5+ years of ML software experience with frameworks including Pytorch, Tensorflow, Jax/Flax

Nice to have

• Experience with ML models for robotics or self-driving vehicles
• Strong C++ in production, integrating ML models
• Training large-scale models on GPU/TPU clusters

Culture & Benefits

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