Staff Research Scientist

Who you are

• PhD or Masters in Computer Science, Machine Learning, Robotics, or a similar technical field; with 3+ years of industry or post-doc research experience in Data-centric AI, Reinforcement Learning or Foundation Models
• Demonstration of original contributions to the field through high-impact publications (ArXiv, peer-reviewed conferences like NeurIPS/ICLR/CVPR), technical blog posts, or significant open-source contributions
• Proficiency and in-depth knowledge of the inner workings of an ML framework (e.g. Pytorch, JAX, Tensorflow)
• Extensive experience working with data and recipes for large scale foundation models
• ML infra experience: training, evaluating and deploying ML models at scale
• Deep learning experience, especially with generative models, e.g., LLMs/VLMs, and/or reinforcement learning

What the job involves

• In this hybrid role, you will report to a Senior Director of AI Foundations
• Own the data recipe for Waymo’s Foundation Model pre-training and post-training
• Lead and drive science on best practices around clustering, filtering, de-duplication, long-tail data mining, memorization, etc
• Tech-lead a team of research engineers to build the data flywheel to enable the above from Waymo’s massive driving data
• Integrate emerging research from the broader community to do rigorous ablations and promising data research techniques. This includes scaling ladders for pre-training, data mix optimizations and RL recipes (preferences) for post-training
• Engage with the wider research community on best practices around data centric evaluation creation and refinement
• Partner with engineering and research teams across Waymo to share recipes, techniques, and post-training best practices to accelerate our collective know-how

Benefits

• Medical, dental, and vision insurance for employees and dependents
• Employee assistance programs focused on mental health
• Personalized workplace adjustments for diverse needs and abilities, including physical, mental, and neurodivergent considerations
• Access to mental health apps
• Onsite wellness centers
• Second medical opinion for you and your loved ones
• Medical advocacy program for transgender employees
• Counseling services
• Support programs including menopause benefit
• Competitive compensation
• Regular bonus and equity performance grant opportunities
• Generous 401(k) and regional retirement plans
• Annual cross-company compensation review and pay equity analysis
• 1-on-1 financial coaching
• Fertility and growing family assistance
• Parental leave and baby bonding leave
• Caregiver leave
• Elder care and support
• Backup childcare
• Survivor income benefit
• Paid time off, including vacation, bereavement, sick leave, parental leave, disability, and holidays
• Jury duty leave
• Military leave
• Hybrid work model with remote work opportunities also available
• Educational reimbursement
• Peer learning and coaching platform
• Donation matching programs
• Employee resource groups
• Volunteer hours
• Internal community groups and local culture clubs
• Inspiring spaces to work, recharge, and collaborate with fellow Waymonauts
• On-site meals and snacks
• Fitness centers, massage programs, and ergonomic support
• On-demand fitness, wellbeing, and cooking classes
• Commuter benefits