Waymo

Senior ML Compiler Engineer, Compute

Waymo
Apply
3 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

Waymo develops the Waymo Driver, a full-stack autonomous driving system, and operates the Waymo One robotaxi service that offers paid, driverless rides in select U.S. cities. The company monetizes through ride-hailing fares and partnerships integrating its technology on multiple vehicle platforms. Founded in 2009 as Google’s self-driving project, Waymo is headquartered in Mountain View and is an Alphabet subsidiary, with over 10 million rider-only trips and 100+ million autonomous miles on public roads.

Contact me