Staff Machine Learning Engineer: Autonomy & Perception

Waymo LLC; Mountain View (CA); On-site; USD 251,000 - 310,000; Full time

Benefits offered by this job

Benefits programs

Job summary

Waymo LLC is seeking an experienced ML/Robotics engineer to refine the ML driver, benchmark generative models, and productionize ML systems for autonomous driving. You will collaborate with world-class engineers and PMs to deliver safe and smooth vehicle behaviors on real-world roads.

The role requires an MS in a relevant field and 8+ years of production ML experience, with strong Python/C++ skills and experience in JAX or PyTorch. Mountain View, CA-based, full-time with generous benefits.

Qualifications

• MS in Computer Science, Machine Learning, Robotics, or related technical field, or equivalent practical experience.
• 8+ years of industry experience deploying, evaluating, and maintaining ML-based systems in real-world production environments.
• Proficient programming skills in Python and/or C++, with strong analytical and debugging abilities.
• Expertise with modern deep learning frameworks, such as JAX or PyTorch.

Responsibilities

• Develop next-generation, ML-powered systems that enhance the capabilities of the ML driver and accelerate the rapid scaling of Waymo’s business.
• Fine-tune, adapt, and benchmark state-of-the-art generative models to improve autonomous vehicle perception, motion prediction, and trajectory planning.
• Write high-quality, scalable, and thoroughly tested code to bring cutting-edge ML into production.
• Partner with world-class engineers and product managers to deliver safe and smooth autonomous vehicle behaviors.

Skills

Python

C++

Deep learning

JAX

PyTorch

Debugging

Analytical skills

Education

MS in Computer Science / ML / Robotics

Job description

Waymo LLC is seeking an experienced ML/Robotics engineer to refine the ML driver, benchmark generative models, and productionize ML systems for autonomous driving. You will collaborate with world-class engineers and PMs to deliver safe and smooth vehicle behaviors on real-world roads.

The role requires an MS in a relevant field and 8+ years of production ML experience, with strong Python/C++ skills and experience in JAX or PyTorch. Mountain View, CA-based, full-time with generous benefits.