Machine Learning Engineer, Training

About the Role

Waymo is seeking a Machine Learning Engineer to join our ML Infrastructure team. This team collaborates with Research and Production teams to develop models in Perception and Planning, which are essential to our autonomous driving software. We focus on providing the best solutions for the entire model development lifecycle, working closely with teams at Google to scale models and solve unique ML challenges for autonomous driving.

Responsibilities

Develop infrastructure components for distributed training, including job scheduling, resource management, data distribution, and model synchronization.
Implement automation solutions for provisioning, deployment, monitoring, and scaling of distributed training infrastructure to enhance operations and reliability.
Monitor system health, diagnose and troubleshoot issues, and perform routine maintenance tasks to ensure the reliability of the distributed training infrastructure.
Identify performance bottlenecks and optimization opportunities.
Improve the developer experience and performance of our scalable ML framework.

Requirements

Bachelor's degree in Computer Science, Engineering, or related field, or 2+ years equivalent experience.
Experience with distributed systems principles and building distributed systems for production environments.
Solid Python or C++ skills.
Prior experience with Machine Learning frameworks (e.g., TensorFlow, PyTorch) and distributed training algorithms.
Ability to debug complex distributed systems issues.
Experience communicating updates and resolutions to customers and other partners.

Preferred Qualifications

Practical familiarity with ML accelerator profiling tools to uncover performance bottlenecks.
Familiarity with cloud computing platforms (e.g., AWS, Azure, GCP) and experience deploying and managing distributed systems in cloud environments.
Knowledge of optimization and deep learning algorithms.

About Waymo

Waymo is an autonomous driving technology company with the mission to be the most trusted driver. Since its inception 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 mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver powers Waymo One, a fully autonomous ride-hailing service, and can also be applied to a range of vehicle platforms and product use cases. The Waymo Driver has provided over one million rider-only trips, enabled by its experience autonomously driving tens of millions of miles on public roads and tens of billions in simulation across 13+ U.S. states.

Join us in our journey to improve mobility and save lives with cutting-edge technology.

Benefits
Extracted with AI

Discretionary annual bonus program
Equity incentive plan
Generous company benefits program

Similar jobs

Last update: 23 minutes ago

Waymo

Software Engineer, Reasoning Foundation

Join Waymo as a Software Engineer in Reasoning Foundation to develop cutting-edge autonomous driving technology.

8 days ago