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

• Lead the effort of model-platform co-optimization for Waymo's perception foundation model
• Dive deep to training and inference efficiency on various platforms such as TPU, GPU, and CPU
• Collaborate with other teams, such as perception, simulation, and ML infrastructure, to drive the model optimization
• Be curious, innovative, and productive

Requirements

• 5+ years of experience designing and optimizing neural networks using Transformers and ConvNets
• Experience optimizing model efficiency for either large-scale training or inference
• Ability to learn new models and new frameworks
• Experience optimizing ML models

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.