Senior ML Researcher – Prediction & Planning

SupportFinity™; San Francisco (CA); Hybrid; USD 204,000 - 259,000; Full time

Benefits offered by this job

Alphabet-wide benefits

Job summary

Waymo is seeking a Research Scientist in Prediction & Planning on a hybrid basis to advance ML-driven prediction and planning for the Waymo Driver. You will work with a world-class team to translate complex driving data into robust models and behaviors.

The role requires 5+ years in ML or AVs, strong Python/C++ skills, and experience with TensorFlow, JAX, or PyTorch. Collaboration with researchers and product teams is essential.

Qualifications

• BS in Computer Science, ML, Robotics, or equivalent practical experience.
• Demonstrated contributions to the ML community via publications or open-source work.
• Hands-on experience with modern deep learning libraries (TensorFlow, JAX, PyTorch).
• Proficient programming skills (Python, C/C++).
• 5+ years of experience in ML modeling and/or autonomous vehicles.

Responsibilities

• Develop the next-generation ML-powered prediction and planning system for the ML driver.
• Frame real-world challenges as well-defined ML problems; research and apply cutting-edge ML techniques.
• Collaborate with researchers, engineers, and product owners to ensure safe, smooth planning behaviors.
• Develop and evaluate large models and integrate them into production planning software.

Skills

Python

C/C++

Analytical skills

Education

Bachelor's degree in Computer Science or equivalent

Tools

TensorFlow

JAX

PyTorch

Job description

Waymo is seeking a Research Scientist in Prediction & Planning on a hybrid basis to advance ML-driven prediction and planning for the Waymo Driver. You will work with a world-class team to translate complex driving data into robust models and behaviors.

The role requires 5+ years in ML or AVs, strong Python/C++ skills, and experience with TensorFlow, JAX, or PyTorch. Collaboration with researchers and product teams is essential.