Machine Learning Engineer / Applied Scientist, Prediction & Planning

About this role

Waymo seeks a Machine Learning Engineer/Applied Scientist to develop state-of-the-art predictive and planning models that enable autonomous vehicles to navigate complex environments. You'll translate real-world driving challenges into ML problems, deploying generative models and reinforcement learning techniques at scale.

What you'll do

• Design, implement, and evaluate generative models for autonomous vehicle planning and prediction
• Develop ML-powered systems that enhance the Waymo Driver's capabilities and support business scaling
• Convert open-ended driving challenges into well-defined machine learning problems using cutting-edge techniques
• Write production-quality, scalable, and thoroughly tested code to transition research into deployment
• Collaborate with researchers, engineers, and product managers to deliver safe planning behaviors
• Publish research findings at top-tier academic venues

What they're looking for

• Deep learning frameworks (JAX, PyTorch)
• Python and/or C++ programming
• Reinforcement learning
• Generative modeling
• Motion planning and prediction algorithms
• Machine learning systems design and evaluation
• Analytical and debugging skills
• Foundation models and causal reasoning

Benefits

• Discretionary annual bonus program
• Equity incentive plan
• Generous company benefits package
• Hybrid work arrangement
• Opportunity to publish at top-tier venues
• Collaboration with world-class researchers