Staff Software Engineer, Waymo

Simulation ML Infrastructure

£145-157k

Waymo employees are also eligible to participate in Waymo’s discretionary annual bonus program and equity incentive plan

Tensorflow

PyTorch

Senior and Expert level

London

Autonomous driving technology

Open for applications

Autonomous driving technology

1001+ employees

B2CTransportAutomation

Open for applications

£145-157k

Waymo employees are also eligible to participate in Waymo’s discretionary annual bonus program and equity incentive plan

Tensorflow

PyTorch

Senior and Expert level

London

1001+ employees

B2CTransportAutomation

Company mission

To make it simpler and safer for people to move things and themselves around.

Job

Company

Role

Who you are

8+ years of professional software engineering experience, with at least 5 years in machine learning infrastructure such as developing, scaling, training, deploying, and optimizing large-scale machine learning systems from data to model
Solid experience in the development and optimization of machine learning infrastructure tools like DeepSpeed, PyTorch, TensorFlow, Ray, or similar frameworks
Expertise in distributed training techniques, including gradient sharding and optimization strategies for scaling large models across ML accelerator profiling tools to uncover performance bottlenecks. Familiarity with custom kernels for compute based efficiency
Experience with state-of-the-art machine learning models such as autoregressive transformers
With experience navigating cross-functional teams and providing technical leadership projects across multiple organizations with the ability to translate complex technical concepts for a broad audience
Practical familiarity in Autonomous Driving, Simulations, and ML accelerators is a plus

What the job involves

We are looking for an experienced Staff Machine Learning Infrastructure Engineer to lead the development of advanced AI/ML infrastructure for multi-billion parameter foundation models in ML accelerator-friendly simulations
Your expertise in massive model scaling, ML accelerators, and distributed training will be required for designing and scaling our systems
This role reports to an Engineering Manager
Be part of a world-class, research engineering team to improve the of ultra realistic multi-agent simulations using foundation models
Collaborate with the core Waymo Realism Modeling team in London and Waymo Oxford to use large foundation models to improve sim realism
Provide deep technical leadership on large-scale ML model architectures, especially for autonomous vehicle models
Work at the intersection of data engineering, model development, and simulations, and provide guidance on architectural decisions and technical directions
Manage complex systems, driving architectures that meet technical and goals
Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation, model training, and evaluation
Collaborate to derive performance and system-level requirements for large ML systems
Translate product/business goals into measurable technical deliverables, ensuring system component agreement
Mentor junior engineers, growing their expertise and promoting a collaborative culture

Salary benchmarks

Our take

Waymo is an autonomous driving technology company that began as the Google self-driving car project in 2009. After initial development, it secured permits to deploy autonomous ride-hailing services in cities including Phoenix and San Francisco, but has recently hit a major bump in the road with safety concerns that resulted in a recall of over 400 self-driving cars.

With the autonomous driving industry already besieged by setbacks and public distrust, this development does Waymo no favours. That said, as an industry leader there remains the possibility it will see out the storm, but massive attention will need to be given to rebuilding trust with consumers if this is to be the case.

Steph

Company Specialist at Welcome to the Jungle

Insights

Led by a woman

Top investors

-1% employee growth in 12 months

Glassdoor (3.7)

Company

Funding (last 2 of 4 rounds)

Aug 2024

$5.6bn

SERIES C

Jun 2021

$2.5bn

GROWTH EQUITY VC

Total funding: $11.1bn

Company HQ

Mountain View, CA

Leadership

Dmitri Dolgov

(Co-CEO, not founder)

Former CTO of the company. After a PhD at the University of Michigan, they worked as a Stanford Postdoctoral Researcher before moving to the Toyota Research Institute. They also spent 8 years at Google as a Software Engineer.

Tekedra N. Mawakana

(Co-CEO, not founder)

Has worked at Waymo since 2017, as Global Head of Policy then CXO then COO before becoming Co-CEO. They are on the Board of Advisors at Operator Collective, Board of Directors at Intuit, and an Advisor for Boom Supersonic.

Share this job

View 60 more jobs at Waymo