Machine Learning Infrastructure Engineer - Remote

Overview

Waymo is an autonomous driving technology company with the mission to be the 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 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.

In Short

• Develop the infrastructure components necessary for distributed training
• Implement automation solutions for provisioning, deployment, monitoring, and scaling of distributed training infrastructure
• Monitor system health, diagnose 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 4+ years equivalent experience
• Experience 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

Benefits

• Waymo employees are also eligible to participate in Waymo’s discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements.