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 large-scale and efficient inference solutions for Waymo.
• Focus on the development of Waymo Foundation Models for driving.
• Work on evaluation, auto-labeling, and distillation.
• Collaborate with Research and Production teams.
• Offer best-in-class solutions for the model development lifecycle.
• Optimize scalable inference platforms for real-time and batch processing.
• Use profiling tools to uncover performance bottlenecks.
• Participate in software development projects using C++ or Python.
• Deploy solutions on cloud platforms like AWS, Azure, or GCP.
• Report to an Engineering Manager.

Requirements

• B.S. in Computer Science, Math, or equivalent real-world experience.
• Solid understanding of machine learning fundamentals.
• Experience with popular ML frameworks such as JAX, PyTorch, or TensorFlow.
• Practical familiarity using profiling tools.
• Experience in software development using C++ or Python.

Benefits

• Eligible for Waymo’s discretionary annual bonus program.
• Participate in equity incentive plan.
• Generous Company benefits program.