Staff ML Engineer, Foundation Models

The mission of the Waymo AI Foundations team is to develop machine learning solutions addressing open problems in autonomous driving, towards the goal of safely operating Waymo vehicles in dozens of cities and under all driving conditions. As part of our work, we also initiate and foster collaborations with other research teams in Alphabet. AI Foundations areas that we are currently focusing on include reinforcement learning, learning from demonstration, generative modeling, Bayesian inference, hierarchical learning, and robust evaluation.

This role follows a hybrid work schedule and reports to a Director of AI Foundations. 

You will:

• Work on connecting large-scale foundation models with production systems.
• Adapt our foundation models to new sensors, new platforms, and new production requirements.
• Collaborate extensively with other teams to land foundation models to our next-gen platforms.

• Experience in building large-scale systems
• Track record of solving complex system problems
• Track record of successful deliveries through large-scale collaborations

Nice to have

• Experience in LLM/VLM related foundation models or systems
• Experience with foundation model post-training