Waymo
Senior Research Scientist, AI Foundations Recipes
Oxford, UK ; London, UK. Posted 3 months ago
The role
You will partner with foundation model training teams to determine data mixtures, training curricula, and design choices. You will design global-scale pipelines and develop data-selection algorithms to improve model capabilities for the Waymo Driver.
What you will do
- Partner with foundation model training teams to determine optimal data mixtures, training curricula, objectives, and other design choices
- Design global-scale pipelines for dynamically updating and reusing data mixtures
- Develop and deploy principled data-selection algorithms
- Run large-scale empirical studies linking pre-training data composition to downstream post-training performance
- Collaborate closely with behavior, perception, and evaluation teams to test foundation models
Required qualifications
- 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 have
- 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
Benefits
- Waymo 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. 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.