Software Engineer, Machine Learning

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.

In this hybrid role, you will report to a Senior Research Scientist.

You will:

• Participate in Waymo’s foundation model post-training and evaluation
• Develop cutting edge RL and Distillation techniques for Autonomous Vehicle Trajectory Planning
• Integrate emerging research from the broader AI community into Waymo’s internal infrastructure, conducting rigorous ablations to identify and scale the most promising methods
• Collaborate with other teams to share techniques developed in the AI Foundations teams to other teams at Waymo
• Integrate Waymo’s multimodal models into simulators of different levels of fidelity

You have:

• Bachelor degree in Computer Science, similar technical field of study, or equivalent practical experience
• Proficiency in writing and debugging python/numpy-style code
• A willingness to work with complexity of globally distributed inference infrastructure

We prefer:

• Experience with large scale (many-machine) training infrastructure and techniques for inference with large models such as model sharding/tensor-parallel
• Reinforcement Learning infrastructure experience
• Large scale distributed inference experience