Staff Research Scientist, Foundation Models Recipes

What you'll do

• Lead research on data clustering, filtering, de-duplication, long-tail data mining, and memorization.
• Tech-lead research engineers building a data flywheel from Waymo’s driving data and conduct rigorous experiments on data and training techniques.
• Collaborate across Waymo and with the research community on data-centric evaluation and post-training practices.

What they ask for

• PhD or Master’s in Computer Science, Machine Learning, Robotics, or a similar technical field.
• 3+ years of industry or postdoctoral research experience in Data-centric AI, Reinforcement Learning, or Foundation Models.
• Evidence of original contributions through high-impact publications, technical blog posts, or significant open-source contributions.
• In-depth proficiency with an ML framework such as Pytorch, JAX, or Tensorflow, and deep learning experience, especially with generative models and/or reinforcement learning.

Nice to have

• Extensive experience with data and recipes for large-scale foundation models.
• ML infrastructure experience training, evaluating, and deploying models at scale.

Pay

$251,000 — $310,000 USD (base salary)

Tools

Pytorch, JAX, Tensorflow

AI focus

Research and develop data recipes for foundation-model pre-training and post-training, including generative models, reinforcement learning, data optimization, and data-centric evaluation.