Machine Learning Engineer, Simulation
Mountain View, California$213k - $263kFull-TimePosted 5d ago
About the role
Waymo is hiring a Machine Learning Engineer for the Simulation team to build ultra-realistic world simulations for testing and training the Waymo Driver, applying large-scale ML models and reinforcement learning to embodied agents, multimodal world models, and counterfactual simulation scenarios.
Responsibilities
- Drive innovation in ultra-realistic world simulation for autonomous vehicles by developing and scaling foundation models for embodied agents.
- Design and train multimodal world models and generative architectures using diverse sensory data (such as RGB images, video, and text).
- Drive closed-loop training and reinforcement learning pipelines to advance simulation realism, policy optimization, and decision-making for autonomous driving.
- Work with Waymo's foundational AI research team to transfer research into scalable, production-ready solutions.
Qualifications
- 5+ years of experience with ML software engineering
- 5+ years of experience with Python
- 5+ years of ML software experience with machine learning frameworks including PyTorch, TensorFlow, or JAX/Flax
- Demonstrated experience building, training, or scaling world models or multi-modal models on large-scale datasets (e.g., video, images, text)
- Hands-on experience with closed-loop training, policy learning, or reinforcement learning frameworks
Skills
Required:PythonML Frameworks (PyTorch, TensorFlow, or JAX/Flax)Multimodal / World ModelsReinforcement Learning / Closed-loop Training
Benefits
- Waymo employees are eligible to participate in Waymo’s discretionary annual bonus program, equity incentive plan, and generous Company benefits program (subject to eligibility)
Last updated Oct 22, 2026. Interested? Apply on the company site.
About Waymo
The World’s Most Experienced Driver
Waymo—formerly the Google self-driving car project—makes it safe and easy for people & things to get around with autonomous vehicles.
Autonomous DrivingAutonomous VehiclesMobilityRide-Hailliquid-ai-recruiting-target
More open roles at Waymo
- Staff Machine Learning Engineer, TLM , Labeling Automation5d agoMountain View, California · Software Engineering
- Technical Program Management, Commercialization5d agoMountain View, California, San Francisco, California · Product
- Technical Program Manager5d agoRemote · Waymo
- Staff Systems Engineer, Driving Behaviors5d agoMountain View, California, San Francisco, California · Systems Engineering
- Lead Program Manager - New Platform Validation Operations5d agoMountain View, California, San Francisco, California · Operations