Waymo
Staff/Sr. Staff Machine Learning Engineer - World Modeling, Simulation
$251K–$310K
Mountain View, CA, USA +1 · Senior Staff · Full-time · Posted Mar 6, 2025
- Location
- Mountain View, CA, USA
Mountain View, California, United States - Experience
- 8+ years of industry or relevant research lab experience, or tenure-track/associate professorship equivalent.
- Education
- PhD or MS in Computer Science, Machine Learning, Robotics, Computer Vision, or a related quantitative field (or equivalent practical research/industry tenure, including academic appointments looking to transition to industry).
About the job
Leads development of large-scale generative world models and simulation systems for training, testing, and evaluating autonomous driving technology.
What you'll do
- Set the technical roadmap for 4D world modeling, generative sensor simulation, and related architectures.
- Train and post-train multimodal world models, and scale inference and evaluation for closed-loop simulation.
- Partner across research and engineering teams, mentor senior engineers, and drive production deployment.
What they ask for
- Demonstrated experience training large-scale generative or physical world models from scratch on multi-node GPU/TPU clusters.
- Hands-on experience with generative-model post-training and distillation, including SFT and reinforcement learning alignment.
- Deep technical foundation in 3D/4D computer vision and physical world modeling.
- Expert proficiency in Python and modern distributed ML frameworks.
Nice to have
- Experience with non-RGB sensor simulation, such as LiDAR and imaging radar.
- Experience in embodied AI and world dynamics, high-performance infrastructure, or modern C++.
Pay
$251,000–$310,000 USD base salary
Tools
Python, JAX/Flax, XLA/Pallas, PyTorch, Megatron-LM, Pathways, DeepSpeed, C++
AI focus
Develops foundation world models for realistic, controllable autonomous-driving simulation, including multimodal pre-training, post-training and alignment, sensor generation, and closed-loop evaluation.
Model-written summary · the original posting is authoritative