Waymo

Senior ML Compiler Engineer, Compute

Waymo
Apply
5 months ago
Bengaluru, IndiaSenior
H1B Sponsor

Responsibilities

  • Analyze performance characteristics of production compiler-generated code and implement optimizations.
  • Design and implement compiler support for novel features of Waymo’s high-performance architecture.
  • Collaborate with hardware architects on hardware/software codesign for a neural-network inference platform.
  • 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 in either field with 3+ years of industry experience; or a PhD in either field with 1+ years of industry experience.
  • 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.
  • Python programming experience is preferred.
  • Experience with compilers for neural networks is preferred.
  • Knowledge of computer architectures used for neural-network inference 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.
  • Some work hours must align with the Pacific Time zone as needed.
  • Eligibility for a 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.