Waymo Logo

Waymo

Sr. Machine Learning Engineer

Reposted 14 Days Ago
Be an Early Applicant
In-Office
London, Greater London, England
Senior level
In-Office
London, Greater London, England
Senior level
The job involves developing machine learning models, enhancing evaluation systems, collaborating with teams, and overseeing production and optimization of models for autonomous vehicles.
The summary above was generated by AI

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 now 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. The Waymo Driver 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 15+ U.S. states.

About the Team The DUE ML Core London team builds and operates scalable machine learning systems, simulation workflows, and insight tools designed to improve the evaluation and developer onboarding journeys. By combining expert human judgment with advanced machine learning models, we deliver training and evaluation data for hundreds of metrics and components that comprise the Waymo Driver. We are looking for researchers and software engineers passionate about developing ML techniques for evaluation systems and driving performance improvements across our technology stack.

You will:

  • Build scalable systems for training and fine-tuning large-scale generative models to produce and evaluate realistic driving behaviors.
  • Lead the implementation and iteration of novel Reinforcement Learning (RL) algorithms, reward functions, and training paradigms tailored for generating high-fidelity driving behaviors.
  • Lead the development of cutting-edge Deep Learning and Generative AI (LLM/VLM) solutions to enhance human-led triaging, automate high-volume workflows, and detect critical anomalies in driving behavior.
  • Oversee the production and optimization of ML models used to assess the performance of Waymo's fleet across millions of miles.
  • Monitor industry trends and Alphabet-wide research to develop novel Reinforcement Learning from Human Preference (RLHF) based data collection and evaluation systems.
  • Partner with Prediction, Planning, and Research teams, as well as senior leadership, to deliver on important strategic efforts.

We'd like you to have:

  • M.S. or Ph.D. in Computer Science, Machine Learning, AI, or a related technical field (or equivalent practical experience).
  • 5+ years of hands-on experience applying Machine Learning models, with a specific focus on Reinforcement Learning.
  • Demonstrated expertise in deep learning, sequence modeling, and generative models.
  • A strong publication record or a history of impactful project delivery in RL or related areas.
  • Proficiency in Python and standard ML frameworks (e.g., JAX, TensorFlow).
  • Experience with large-scale distributed training and data processing.

We prefer you to have:

  • 7+ years of relevant experience in ML/RL research and application.
  • Experience in autonomous vehicles, robotics, or complex simulation environments.
  • Familiarity with state-of-the-art RL techniques, specifically for fine-tuning large models (e.g., RLHF).
  • Experience integrating large-scale simulation platforms with ML training workflows.
  • A track record of technical leadership and influencing senior stakeholders.

The expected base salary range for this full-time position is listed below. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level.  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.

Salary Range
£120,000£130,000 GBP

Top Skills

C++
Hugging Face'S Transformers
Python
PyTorch
TensorFlow

Similar Jobs

19 Days Ago
Hybrid
Cambridge, Cambridgeshire, England, GBR
Senior level
Senior level
Artificial Intelligence • Semiconductor
Develop and optimize machine learning models tailored for AI hardware, collaborate with cross-functional teams, and conduct experiments to improve efficiency.
Top Skills: C++CudaHpcJaxKubernetesMachine LearningPythonPyTorchTriton
19 Days Ago
Hybrid
Bristol, England, GBR
Senior level
Senior level
Artificial Intelligence • Semiconductor
As a Senior Machine Learning Engineer, you will develop and optimize AI models for specialized hardware, scale systems, and collaborate with cross-functional teams.
Top Skills: C++CudaInfinibandJaxKubernetesNvlinkPythonPyTorchRoceTriton
19 Days Ago
Hybrid
London, Greater London, England, GBR
Senior level
Senior level
Artificial Intelligence • Semiconductor
Develop and optimize AI models for high-performance systems, collaborate with cross-functional teams, evaluate new methods, and improve model efficiency.
Top Skills: C++CudaInfinibandJaxKubernetesNvlinkPythonPyTorchRoceTriton

What you need to know about the London Tech Scene

London isn't just a hub for established businesses; it's also a nursery for innovation. Boasting one of the most recognized fintech ecosystems in Europe, attracting billions in investments each year, London's success has made it a go-to destination for startups looking to make their mark. Top U.K. companies like Hoptin, Moneybox and Marshmallow have already made the city their base — yet fintech is just the beginning. From healthtech to renewable energy to cybersecurity and beyond, the city's startups are breaking new ground across a range of industries.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account