Senior Machine Learning Engineer - Model Optimization - 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

• Work across the entire ML framework/compiler stack (e.g. JAX, XLA, Triton, and CUDA), and system-efficient deep learning models.
• Deep dive into the whole stack of ML software stack, from custom ops, framework/ ML compiler, to low-level libraries.
• Apply model optimization and efficient deep learning techniques to models and optimized ML operator libraries.

Requirements

• M.S. in CS, EE, Deep Learning or a related field
• 3+ years of experience on model optimization or efficient deep learning techniques
• Strong Python or C++ programming skills
• Solid experience with designing, training and debugging deep learning models to achieve the highest scores/accuracies.

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.