Staff Software Engineer/Data Scientist, Large Model Evaluation

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver powers Waymo’s fully autonomous ride-hail service and can also be applied to a range of vehicle platforms and product use cases. The Waymo Driver has provided over ten million rider-only trips, enabled by its experience autonomously driving over 100 million miles on public roads and tens of billions in simulation across 15+ U.S. states.

The Large Model Evaluation team is at the nexus of Waymo’s AI ambition. With advancements in Large Language Models (LLMs) and Vision-Language Models (VLMs), Waymo is building state-of-the-art AI systems that handle the full complexity of real-world driving. At its core, our progress is defined by our ability to measure it. While robust evaluation is the bottleneck for deploying any large model, the challenge at Waymo is uniquely complex and safety-critical. We are looking for quantitatively-minded engineers to research and propose new ways to assess the ML models deployed in the Waymo Driver.

You will:

• Develop novel metrics and sampling techniques to measure the driving trajectories generated by ML models.

• Employ creative simulation strategies to measure the driving performance of generative AI models. Identify potential edge cases, and provide reliable performance insights that inform model development and deployment.

• Build data pipelines for signal discovery, data labeling, feature extraction and metric computation based on large-scale simulations.

• Conduct data analysis to diagnose regressions in ML models.

• Collaborate with world-class engineering and research teams that develop large-scale ML models.

You have:

• 7+ years of relevant industry experience in a heavily quantitative software engineering area

• Experience navigating complex technical and product landscapes, defining technical strategy, and creating roadmaps.

• Software Engineering Fundamentals:

• Proficiency in programming in Python or C++

• Experience with software design principles, coding best practices, testing methodologies, and version control software.

• Experience building software pipelines for data processing, system evaluation, or metric computation, in the context of large-scale systems.

• Machine learning & Quantitative Experience

• Knowledge of AI fundamentals, such as transformer architectures, distillation techniques, etc.

• Experience evaluating the quality of ML models

• Demonstrated experience taking quantitative findings through to productionized tools.

We prefer:

• Experience with simulation systems, robotics, or autonomous vehicles.

• Familiarity with one of the modern deep learning frameworks (e.g. JAX, Tensorflow)

• Experience leading a team of Engineers

The expected base salary range for this full-time position across US locations is listed below. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level. Your recruiter can share more about the specific salary range for the role location or, if the role can be performed remote, the specific salary range for your preferred location, during the hiring process.

Waymo employees are also eligible to participate in Waymo’s discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements.

Salary Range
$238,000—$302,000 USD