The opportunity
Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to…
What you'll do
Understand the problem the current LPM model is trying to solve along with its details
Improve/extend the current RL infrastructure to support LPM model training needs
Profiling and understanding slowness, maintain and improve the training efficiency
Identify promising new RL technologies and integrate to our infrastructure
Currently pursuing a PhD/Master degree in Computer Science, Electrical: Engineering, Machine Learning, Robotics, or a related technical field
Fundamental understanding of machine learning, deep learning and reinforcement learning
What they're looking for
- Familiar with modern frameworks (JAX, Flax, PyTorch, or TensorFlow)
- Strong software development skills with Python
- Has experience on ML or RL with hands-on projects
- Has experience on large scale ML system infrastructure
