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

Nice to Haves:

• Proven track record of deploying ML models to production environments
• Deep learning experience, especially with generative models, e.g., LLMs/VLMs, and/or reinforcement learning or imitation learning, transformer models, autoencoders and embeddings
• Autonomous driving experience
• C++ experience

What You'll Be Doing:

• 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.

Perks and Benefits:

• Expected base salary range: £123,000 - £129,000 GBP
• Eligibility for Waymo’s discretionary annual bonus program
• Equity incentive plan
• Generous Company benefits program, subject to eligibility requirements