Waymo

Senior ML Compiler Engineer, Compute

Waymo
Apply
2 months ago
Mountain View, CA, USA or New York, NY, USASenior
H1B Sponsor

Base Salary

$213k - $263k/yr

Responsibilities

  • Leverage agentic AI to accelerate analysis, planning, and implementation of production software improvements.
  • Analyze performance characteristics of compiler-generated code and implement optimizations to improve performance.
  • Design and implement compiler support for novel features of Waymo’s high-performance architecture.
  • Collaborate with hardware architects on the development of a neural-network inference platform through hardware/software codesign.
  • Work with model developers to tune neural networks for improved inference efficiency and accuracy.

Requirements

  • BS degree in Computer Science or Electrical Engineering, or equivalent experience, with 5+ years of industry experience; or an MS degree with 3+ years; or a PhD degree with 1+ years.
  • At least 3 years of experience working on compilers for parallel architectures.
  • At least 1 year of experience working with ML inference or linear algebra computation.
  • C++ programming skills, including leveraging agentic AI acceleration.
  • Python programming experience is preferred.
  • Experience with neural-network compilers, especially MLIR-based compilers, is preferred.
  • Knowledge of neural-network inference computer architectures and neural-network performance characteristics is preferred.
  • Knowledge of principles behind popular machine-learning and neural-network algorithms and applications is preferred.

Benefits

  • Hybrid work schedule
  • Eligibility for discretionary annual bonus program, equity incentive plan, and company benefits program

Tech Stack

Categories

Waymo

About Waymo

1,001-5,000 employees

On the journey to be the world's most trusted driver. With the Waymo Driver, we can improve mobility while saving thousands of lives. Download the Waymo One app and ride today. Waymo reaches out to candidates from official channels only (e.g. directly from @waymo.com email addresses, or through our recruiters or sourcers who are noted as such on LinkedIn). We do not contact candidates about career opportunities through instant messaging apps like Telegram, email addresses from domains other than waymo.com (such as Gmail addresses), direct messages on Twitter, Facebook, and Instagram, or text messages. Visit waymo.com to check out our official job listings. Need support? Send us a message – we're here to help.

Contact me