Senior ML Engineer - Runtime & Optimization, Machine Learning Platform

Key Skills

C++, CUDA/GPU programming, Deep Learning, Machine learning techniques, Optimization, Python Programming

Job Description

The ML Platform team at Waymo provides a set of tools and technologies to support and automate the lifecycle of the machine learning workflow, including feature and experiment management, model development, debugging & evaluation, deployment, and monitoring. These efforts have resulted in making machine learning more accessible to teams at Waymo, including Perception, Behavior Prediction, Planner, Routing, Maps and Research, ensuring greater degrees of consistency and repeatability, and addressing the “last mile” of getting models into production and managing them once they are in place. We work hand in hand with machine learning experts in all parts of the company as well as our collaborators across Alphabet.

We are looking for engineers with an ML compiler, runtime, or GPU background to help us improve compute performance on our car. You’ll work across the entire ML compiler stack (e.g. TF, XLA, MLIR, and CUDA). You will be pleasantly challenged with deploying a hugely diverse set of Waymo ML models on limited computation resources. Non-exhaustive examples of the types of work you will work on:

• Collaborate with ML practitioners to understand their models and modify model architectures to run faster on the car.
• Build tools to benchmark and analyze workloads.
• Build tools to productize deep learning models.

At a minimum, we’d like you to have:

• B.Sc in Computer Science, Mathematics or a related field
• 5+ years of industry experience
• Strong C++ programming skills
• Passion for developing and optimizing compilers for modern architectures (MLIR etc.)
• Working knowledge of compiler architecture front-end and middle-end optimizations, scheduling, register allocation, and back-end code generation

It’s preferred if you have:

• M.Sc or PhD in Computer Science, Mathematics or a related field
• Strong Python programming skills
• Experience with efficient deep learning techniques such as quantization, NAS, distillation
• Experience with any of ML compiler and IRs (HLO, MLIR, Relay)

The expected base salary range for this full-time position across US locations is listed below. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level. Your recruiter can share more about the specific salary range for the role location or, if the role can be performed remote, the specific salary range for your preferred location, during the hiring process. 

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. 

Salary Range

$187,000—$233,000 USD

While at Waymo, you will enjoy benefits that cover…

Health and wellness: Our people are at the heart of everything we do. At Waymo, you can enjoy top-notch medical, dental and vision insurance, mental wellness support, a Flexible Spending Account (FSA), a Health Saving Account (HSA), on-site physicians and/or nurses in some locations, and special wellness programs.

Financial wellness: Your financial peace of mind is important to us. At Waymo, we offer competitive compensation, bonus opportunities, equity, a generous 401(k) plan or regional retirement plans, 1-on-1 financial coaching, a 529 College Savings Plan and lots of other perks and employee discounts.