ML Engineer, Autonomous Driving & Planning

Waymo; California (MO); Hybrid; USD 175,000 - 215,000; Full time

Benefits offered by this job

Bonus program

Equity incentives

Benefits package

Job summary

Waymo is seeking a highly skilled researcher to design, implement, and evaluate state-of-the-art generative models for autonomous vehicle planning and prediction.

You will develop ML-powered systems that enhance the Waymo Driver and accelerate scaling of the business, translating real-world driving challenges into well-defined ML problems using advanced techniques including foundation models and reinforcement learning.

Qualifications

• PhD in Computer Science, Machine Learning, Robotics or related field, or equivalent practical experience.
• Hands-on expertise with modern deep learning frameworks (e.g., PyTorch, JAX).
• Strong programming skills in Python and/or C++ with good debugging abilities.
• Publications in top-tier ML/AI conferences or significant research impact.

Responsibilities

• Design, implement, and evaluate state-of-the-art generative models for autonomous vehicle planning and prediction.
• Develop ML-powered systems to enhance the Waymo Driver and scale capabilities.
• Translate real-world driving challenges into well-defined ML problems using foundation models and RL.
• Write high-quality, scalable production code and collaborate with researchers, engineers, and product managers.

Skills

Python

C++

PyTorch

JAX

Education

PhD in CS / ML / Robotics

Job description

Waymo is seeking a highly skilled researcher to design, implement, and evaluate state-of-the-art generative models for autonomous vehicle planning and prediction.

You will develop ML-powered systems that enhance the Waymo Driver and accelerate scaling of the business, translating real-world driving challenges into well-defined ML problems using advanced techniques including foundation models and reinforcement learning.