Logo
Waymo

Software Engineer, ML Efficiency

Waymo, Mountain View, California, us, 94039

Save Job

Overview Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self‑Driving Car Project in 2009, Waymo has focused on building the Waymo Driver – the world's most experienced driver – to improve access to mobility while saving thousands of lives lost to traffic crashes. The Waymo Driver powers Waymo's fully autonomous ride‑hail service and can also be applied to a range of vehicle platforms and product use cases. It has provided over ten million rider‑only trips, enabled by its experience autonomously driving over 100 million miles on public roads and tens of billions in simulation across more than 15 U.S. states.

Mission The mission of the Waymo AI Foundations team is to develop machine‑learning solutions addressing open problems in autonomous driving, toward the goal of safely operating Waymo vehicles in dozens of cities and under all driving conditions. The team initiates and fosters collaborations with other research teams in Alphabet. Current focus areas include reinforcement learning, learning from demonstration, generative modeling, Bayesian inference, hierarchical learning, and robust evaluation.

Role Summary This role follows a hybrid work schedule and reports to a Principal Research Scientist.

Responsibilities

Enable the team to develop large‑scale end‑to‑end driving models with high hardware efficiency, reliable training and inference, and good developer experience.

Participate in model design from a hardware‑efficiency perspective.

Partner with other Waymo and Google teams to collaborate on infrastructure work.

Qualifications

Bachelor's degree in Computer Science, a similar technical field of study, or equivalent practical experience.

Proficiency in writing and debugging Python/numpy‑style code.

Deep knowledge of the inner workings of an ML framework (e.g., PyTorch, JAX, TensorFlow).

Proficiency in system performance: parallelism, buffering/prefetching, pipelining, and asynchrony.

Preferred

Experience with large‑scale system reliability and failure‑resilience/recovery.

Expertise in ML accelerator programming, including but not limited to CUDA, Triton, or ML compilers.

Knowledge of hardware micro‑architectures and instruction sets.

Proficiency in C++.

Benefits The expected base salary range for this full‑time position across U.S. locations is $204,000 – $259,000 USD. Waymo employees are also eligible to participate in Waymo's discretionary annual bonus program, equity incentive plan, and generous company benefits program, subject to eligibility requirements.

#J-18808-Ljbffr