Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now

About the Role

The AI Foundations team at Waymo is dedicated to creating machine learning solutions for autonomous driving challenges, aiming to enable Waymo vehicles to operate safely in diverse environments. This role involves collaborating with foundation model training teams to optimize data mixtures, training strategies, and objectives, and designing pipelines for dynamic data updates and selection. You will also conduct large-scale studies to link pre-training data composition to downstream performance and collaborate with other teams to evaluate foundation models.

Responsibilities

Partner with foundation model training teams to determine optimal data mixtures, training curricula, objectives, and other design choices that shape model capabilities.
Design global-scale pipelines for dynamically updating and reusing data mixtures as our understanding of model behavior evolves and real-world datasets are continually ingested.
Develop and deploy principled data-selection algorithms (e.g. influence estimation, proxy models, gradient similarity search) to find targeted subsets of data for fine tuning and RL.
Run large-scale empirical studies linking pre-training data composition to downstream post-training performance, translating those scaling laws into optimized production training runs.
Collaborate closely with behavior, perception, and evaluation teams to test these foundation models and measure their impact on the performance of the Waymo Driver.

Requirements

PhD in ML or related in AI/CS, or MSc with a strong academic publication record
Strong statistical / ML knowledge
Proficiency in Python
3+ years of experience with modern deep learning frameworks: JAX, TensorFlow/PyTorch

Skills

Generative models
LLMs/VLMs
Reinforcement learning
Imitation learning
Transformer models
Autoencoders
Embeddings
C++

Location

Hybrid

Work Type

Hybrid
Full-time

Experience Level

3+ years of experience

Education Level

PhD in ML or related in AI/CS
MSc with a strong academic publication record

Salary/Compensations

£123,000—£129,000 GBP

Benefits

Discretionary annual bonus program
Equity incentive plan
Generous Company benefits program

About the Company

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.