ML Engineer, Foundation Model Recipes (AI)

TL;DR

ML Engineer, Foundation Model Recipes (AI): Develops machine learning recipes and evaluation systems for embodied AI agents in autonomous driving with an accent on foundation models, robotics, data, and safety. Focus on building large-scale data and evaluation pipelines, advancing model quality and realism, and transferring research innovations into production.

Location: Mountain View, California, United States; hybrid role with onsite work requirements.

Salary: $175,000–$215,000 USD base salary, plus potential annual bonus, equity, and company benefits.

Company

Waymo develops autonomous driving technology and operates a fully autonomous ride-hail service powered by the Waymo Driver.

What you will do

• Develop and extend machine learning and robotics research for embodied AI agents.
• Work with state-of-the-art foundation models to improve quality, safety, and realism.
• Drive model development through data, evaluation, and systems work.
• Implement and extend large-scale data and evaluation pipelines.
• Collaborate across research and engineering organizations to bring innovative technology into production.

Requirements

• Master’s degree in Computer Science, Machine Learning, Robotics, or a similar technical field, or equivalent practical experience.
• Proficiency in Python.
• Familiarity with a modern deep learning framework such as PyTorch, JAX, or TensorFlow.
• Previous industrial or research experience developing recipes for machine learning models.
• Ability to work in a hybrid, onsite-oriented role in Mountain View, California.

Nice to have

• Publications in top-tier conferences or contributions to leading open-source projects.
• Strong software engineering skills and experience designing large distributed pipelines.
• Experience in autonomous vehicle planning and related research.
• Experience labeling and curating data for machine learning evaluation and training.

Culture & Benefits

• Work on autonomous driving and AI Foundations research targeting safe operation in diverse cities and driving conditions.
• Opportunities to collaborate with research teams across Alphabet.
• Eligibility for a discretionary annual bonus program.
• Equity incentive plan and company benefits, subject to eligibility.