ML Systems Engineer — Runtime & Optimization (Hybrid, Equity)

Waymo; Mountain View (CA); Hybrid; USD 204,000 - 259,000; Full time

Benefits offered by this job

Discretionary annual bonus program

Equity incentive plan

Generous company benefits program

Job summary

An autonomous driving technology company is looking for an experienced engineer to improve compute performance in machine learning systems. This hybrid role involves collaboration with a world-class ML team and requires strong expertise in ML software or systems. The ideal candidate has a Master's degree in a relevant field, with at least 5 years of experience in ML. A competitive salary range of $204,000—$259,000 USD is offered, along with additional benefits.

Qualifications

• 2+ years of experience as a technical lead including writing project plans.
• 5+ years of experience developing solutions in ML systems or ML software stack.
• Familiarity with HW architecture of ML hardware accelerators (e.g., GPU/TPU) preferred.

Responsibilities

• Lead collaboration with Waymo ML scientists to enhance ML workloads.
• Analyze and improve ML system workloads on cloud and self-driving cars.
• Apply model optimization techniques to Waymo's ML systems.

Skills

ML software or ML systems expertise

Strong Python or C++ programming skills

Deep understanding of ML system architecture

Performance analysis

Education

M.S. in CS, EE, Deep Learning or a related field

Tools

Pytorch

JAX

TensorFlow

Job description

An autonomous driving technology company is looking for an experienced engineer to improve compute performance in machine learning systems. This hybrid role involves collaboration with a world-class ML team and requires strong expertise in ML software or systems. The ideal candidate has a Master's degree in a relevant field, with at least 5 years of experience in ML. A competitive salary range of $204,000—$259,000 USD is offered, along with additional benefits.