Taro Logo

Senior ML Engineer, Foundation Model Research

Waymo is an autonomous driving technology company developing the Waymo Driver, providing fully autonomous ride-hailing service and advancing self-driving technology.
$204,000 - $259,000
Machine Learning
Staff Software Engineer
Hybrid
1,000 - 5,000 Employees
5+ years of experience
AI · Automotive

Description For Senior ML Engineer, Foundation Model Research

Waymo, an autonomous driving technology company that evolved from the Google Self-Driving Car Project, is seeking a Senior ML Engineer to join their AI Foundations team. This role offers an exciting opportunity to work on cutting-edge machine learning solutions for autonomous driving technology.

The position involves working with large vision-language models, specifically focusing on extending and fine-tuning models like Gemini for autonomous driving applications. You'll be responsible for building datasets and training infrastructure for specific driving tasks, including end-to-end driving, reinforcement learning driving rationale, spatial reasoning, and semantic understanding.

As part of the AI Foundations team, you'll collaborate on developing machine learning solutions addressing open problems in autonomous driving. The team focuses on various areas including reinforcement learning, learning from demonstration, generative modeling, Bayesian inference, hierarchical learning, and robust evaluation.

The role offers competitive compensation ranging from $204,000 to $259,000 USD, along with comprehensive benefits including medical, dental, and vision insurance, mental wellness support, and various financial benefits. The position is hybrid-based in Mountain View, California, with the flexibility to work from another location for four weeks per year.

This is an excellent opportunity for experienced machine learning engineers who want to make a significant impact in autonomous driving technology. The ideal candidate will have strong experience in large language models, distributed training, and a proven track record in end-to-end product development. The role requires proficiency in C++ and Python, along with at least 5 years of industrial experience.

Working at Waymo means joining a team that's dedicated to improving access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver has already provided over one million rider-only trips and has autonomously driven tens of millions of miles across 13+ U.S. states.

Last updated 2 days ago

Responsibilities For Senior ML Engineer, Foundation Model Research

  • Extend and fine-tune existing large vision-language models (e.g. Gemini)
  • Build datasets and train infrastructure for driving specific tasks
  • Leverage language to model the driving rationale and environment context
  • Collaborate with other teams at Waymo to improve Waymo product

Requirements For Senior ML Engineer, Foundation Model Research

Python
  • In-depth knowledge of designing and training large language models and multi-modal applications
  • Experience in end-to-end product development
  • Experience in training generative models and multi-task networks at scale
  • Experience in large-scale distributed training and optimization
  • Experience in large-scale training dataset curation
  • Bachelor's degree or above in relevant fields and 5 years of industrial experience
  • Proficiency in C++ and Python

Benefits For Senior ML Engineer, Foundation Model Research

Medical Insurance
Dental Insurance
Vision Insurance
Mental Health Assistance
401k
Equity
Parental Leave
  • Top-notch medical, dental and vision insurance
  • Mental wellness support
  • Gym membership
  • Special wellness programs
  • Competitive compensation
  • Bonus opportunities
  • Equity
  • Employees provident fund
  • Employee discounts
  • Flexibility to work from another location for four weeks per year
  • Paid time off
  • Bereavement leave
  • Sick leave
  • Parental leave

Interested in this job?

Jobs Related To Waymo Senior ML Engineer, Foundation Model Research