Waymo
Machine Learning Engineer, Simulation Realism (World Model)
$213K–$263K
Mountain View, CA, USA +1 · Full-time · Posted Mar 13, 2025
- Location
- Mountain View, CA, USA
Mountain View, California, United States - Experience
- 5+ years in ML software engineering and 5+ years with Python; 5+ years of ML software experience with listed frameworks.
About the job
Develop and scale multimodal world models and reinforcement-learning pipelines to make autonomous-driving simulations more realistic and steerable.
What you'll do
- Develop foundation models for embodied agents and train multimodal world models using sensory data such as images, video, and text.
- Build closed-loop training and reinforcement-learning pipelines for simulation realism, policy optimization, and autonomous-driving decisions.
- Transfer foundational AI research into scalable, production-ready solutions.
What they ask for
- 5+ years of experience with ML software engineering.
- 5+ years of experience with Python.
- 5+ years of ML software experience with PyTorch, TensorFlow, or JAX/Flax.
- Experience building, training, or scaling world or multimodal models on large-scale datasets, and hands-on experience with closed-loop training, policy learning, or reinforcement-learning frameworks.
Nice to have
- Experience in robotics, control theory, or embodied AI.
- Experience with autonomous-driving ML, production C++, or training large-scale models on GPU/TPU clusters.
Pay
$213,000–$263,000 USD base salary
Tools
Python, PyTorch, TensorFlow, JAX/Flax, C++, GPU, TPU
AI focus
Build multimodal foundation and world models for realistic autonomous-driving simulation, using closed-loop training and reinforcement learning.
Model-written summary · the original posting is authoritative