Waymo logo

ML Compiler Engineer, Compute

Waymo

Role Overview

Waymo is seeking a talented ML Compiler Engineer to join our Compute Team, which is responsible for delivering the compute platform that runs the autonomous vehicle's software stack. This role involves maximizing the performance of our neural networks by enhancing and extending our production-grade compiler. You will work closely with hardware architects and model developers to develop an understanding of our unique neural network inference platform and neural networks.

Responsibilities

  • Implement compiler support for novel features of our high-performance architecture.
  • Collaborate with other team members to ensure the design and improvement of hardware and software for maximum performance.
  • Develop system-level compute architectures that push the boundaries of performance, power, and latency.

Qualifications

  • BS degree in Computer Science/Electrical Engineering or equivalent practical experience and 3+ years of industry experience.
  • MS degree in Computer Science/Electrical Engineering and 1+ years of industry experience.
  • PhD Degree in Computer Science/Electrical Engineering.
  • 1+ years of industry and/or academic experience with compilers and parallel computing.
  • 1+ years of industry and/or academic experience working with ML inference or linear algebra computations.
  • C++ programming skills.

Preferred Skills

  • Python programming experience.
  • Experience with compilers for neural networks.
  • Knowledge of computer architectures used for neural network inference, and neural network performance characteristics.
  • Knowledge of the principles behind popular machine learning and neural network algorithms and applications.

Benefits

  • Participation in Waymo’s discretionary annual bonus program.
  • Eligibility for equity incentive plan.
  • Access to a comprehensive company benefits program.

This is a hybrid role based in New York City, reporting to the Manager of Software Engineering. The expected salary range is $158,000—$200,000 USD, depending on the candidate's location, experience, and skills.

Benefits
Extracted with AI

  • Discretionary annual bonus
  • Equity incentive plan
  • Company benefits program

Similar jobs

Last update: 23 minutes ago

Qualcomm logo
Qualcomm

Machine Learning Compiler Engineer

Join Qualcomm as a Machine Learning Compiler Engineer to optimize ML compilers for cutting-edge accelerators.

Waymo logo
Waymo

Machine Learning Engineer, Training

Join Waymo as a Machine Learning Engineer to develop distributed training infrastructure for autonomous driving.

Waymo logo
Waymo

Software Engineer, Reasoning Foundation

Join Waymo as a Software Engineer in Reasoning Foundation to develop cutting-edge autonomous driving technology.

Waabi logo
Waabi

Remote Software Engineer

Join Waabi as a Remote Software Engineer to develop cutting-edge self-driving technology. Work with AI, Python, C++, and more.

Hop logo
Hop

Machine Learning Engineer - Ads

Join as a Machine Learning Engineer focusing on Ads, developing predictive models in a hybrid role in New York.

micro1 logo
micro1

Machine Learning Engineer with AI/ML Experience

Join us as a Machine Learning Engineer to develop AI/ML models and applications. Work remotely with top-tier companies.

ASML logo
ASML

Machine Learning Mathematics Engineer

Join ASML as a Machine Learning Mathematics Engineer to develop optical metrology solutions using machine learning and optimization algorithms.

Google DeepMind logo
Google DeepMind

Research Engineer, Product

Join Google DeepMind as a Research Engineer to apply cutting-edge AI models to real-world problems. Hybrid work in New York.

Ema Unlimited logo
Ema Unlimited

Machine Learning Engineer

Join Ema Unlimited as a Machine Learning Engineer in SF Bay Area, working on cutting-edge AI solutions with a focus on NLP and ML technologies.

Neural Concept logo
Neural Concept

Machine Learning Engineer

Join Neural Concept as a Machine Learning Engineer to solve complex engineering challenges using Python and deep learning.

PhysicsX logo
PhysicsX

Machine Learning Scientist

Join PhysicsX as a Machine Learning Scientist to develop innovative models using deep learning for physics simulations.

CHAI: AI Platform logo
CHAI: AI Platform

Senior ML Infrastructure Engineer

Join CHAI: AI Platform as a Senior ML Infrastructure Engineer to build and scale ML systems in Palo Alto.

Niantic, Inc. logo
Niantic, Inc.

Senior Software Engineer, Machine Learning

Join Niantic as a Senior Software Engineer in Machine Learning to enhance products using generative AI technologies.

PhysicsX logo
PhysicsX

Machine Learning Engineer

Join PhysicsX as a Machine Learning Engineer to develop innovative models for physics simulations using Python and PyTorch.

Tesla logo
Tesla

Internship Software Engineer - Machine Learning Infrastructure

Join Tesla as an Internship Software Engineer in Machine Learning Infrastructure. Work on AI infrastructure and neural network scaling.

Porsche AG logo
Porsche AG

Machine Learning Engineer for Vehicle Safety Systems

Join Porsche AG as a Machine Learning Engineer to enhance vehicle safety systems using AI and data science.

Adobe logo
Adobe

Machine Learning Engineer - University Graduate 2025

Join Adobe as a Machine Learning Engineer to develop AI technologies for Photoshop and Digital Imaging.

Snap Inc. logo
Snap Inc.

Machine Learning Engineer

Join Snap Inc. as a Machine Learning Engineer in Los Angeles. Develop and deploy ML models to enhance user experience. Competitive salary and benefits.

Olo logo
Olo

Senior Machine Learning Engineer

Join Olo as a Senior Machine Learning Engineer to build and scale ML models for the restaurant industry. Remote work available.

Intel Corporation logo
Intel Corporation

AI Software Development Engineer

Join Intel as an AI Software Development Engineer to develop and deploy AI applications, enhancing engineering productivity.

Argon AI (YC W24) logo
Argon AI (YC W24)

Founding Applied AI Engineer

Join Argon AI as a Founding Applied AI Engineer to lead AI initiatives in pharma, focusing on domain-specific AI and RAG systems.

LlamaIndex logo
LlamaIndex

Founding AI Engineer

Join LlamaIndex as a Founding AI Engineer to shape the future of LLM applications with cutting-edge AI projects.

LlamaIndex logo
LlamaIndex

Founding Applied AI Engineer

Join LlamaIndex as a Founding Applied AI Engineer to build and deploy LLM applications. Competitive salary and equity offered.

LlamaIndex logo
LlamaIndex

Founding AI Engineer, Backend

Join LlamaIndex as a Founding AI Engineer, Backend to build scalable cloud services for LLM applications.