Senior Research Scientist, AI Foundations Recipes
Waymo, Oxford, UK +3
The role
In this hybrid role, you will report to a Senior Staff Software Engineer.
What you will do
- 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.