RemoteLeaf Logo
Log in Sign up

Staff Machine Learning Engineer

4 weeks ago
United Kingdom
Full-time
Lead
Machine Learning Engineer
Software Development
Deep Learning Generative AI LLM Machine Learning Python Reinforcement Learning TensorFlow
Apply Now

Waymo

Waymo is an autonomous driving technology company building the Waymo Driver and operating Waymo One, its fully autonomous ride-hailing service.

Autonomous vehicles, robotics, AI, ride-hailing / mobility tech
Founded 2009
$21600M raised
View All Jobs 72

Description

  • Build scalable systems for training and fine-tuning large-scale generative models to produce realistic and challenging driving behaviors.
  • Lead the implementation and iteration of novel reinforcement learning algorithms, reward functions, and training paradigms.
  • Develop deep learning and generative AI solutions to improve human-led triaging, automate high-volume workflows, and analyze self-driving behavior for critical anomalies.
  • Oversee the production and optimization of machine learning models used to assess Waymo’s fleet of vehicles.
  • Develop a reinforcement learning from human preference (RLHF) based data collection and evaluation system.
  • Monitor and incorporate best practices from Alphabet and the broader industry to improve ML evaluation systems.
  • Collaborate with Prediction, Planning, Research, technical leads, and senior leadership across Waymo on strategic initiatives.
  • Combine expert human judgment with machine learning models to deliver training and evaluation data for metrics and components used by the Waymo Driver.

Requirements

  • M.S. or Ph.D. in Computer Science, Machine Learning, Artificial Intelligence, or a related technical field, or equivalent practical experience.
  • 7+ years of hands-on experience developing and applying machine learning models, with a significant focus on reinforcement learning.
  • Demonstrated expertise in deep learning, sequence modeling, and generative models.
  • Strong publication record or a history of impactful project delivery in reinforcement learning or related areas.
  • Proficiency in Python and standard ML frameworks such as JAX or TensorFlow.
  • Experience with large-scale distributed training and data processing.
  • Proven ability to lead complex and ambiguous technical projects from conception to completion.
  • 10+ years of relevant experience in ML/RL research and application preferred.
  • Experience in autonomous vehicles, robotics, or complex simulation environments preferred.
  • Deep understanding of state-of-the-art RL techniques, including fine-tuning large models from human feedback or preferences preferred.
  • Familiarity with large-scale simulation platforms and integration with ML training workflows preferred.
  • Experience designing and using metrics for evaluating complex AI systems preferred.
  • Track record of technical leadership and influencing senior stakeholders preferred.
  • Excellent communication skills and ability to articulate complex technical concepts clearly preferred.

Benefits

  • £155,000–£163,000 GBP base salary range.
  • Eligibility for Waymo’s discretionary annual bonus program.
  • Eligibility for Waymo’s equity incentive plan.
  • Access to Waymo’s generous company benefits program, subject to eligibility requirements.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

x.ai

ML Infrastructure Engineer

x.ai 51-250 Internet Software & Services

SpaceXAI is hiring an ML Infrastructure Engineer to build and optimize the machine learning platform that powers recommendations on X.

United States Full-time Junior Infrastructure Engineer Machine Learning Engineer
$180k-$440k
Ansible C++ Linux Puppet Python PyTorch Rust
14 hours, 59 minutes ago
Reddit

Senior Machine Learning Engineer, Safety

Reddit 1K-5K Internet Software & Services

Reddit is hiring a remote-friendly Machine Learning Engineer to build and improve safety systems that support enforcement of Reddit rules using large language models.

United States Full-time Senior Machine Learning Engineer
$217k-$303k
Deep Learning LLM Machine Learning NLP Python PyTorch TensorFlow
15 hours, 29 minutes ago
Torc

Senior, Machine Learning Engineer - 3D Perception

Torc 251-1K Road & Rail

Torc is hiring a Senior Machine Learning Engineer – 3D Perception to develop and deploy production perception models for autonomous trucks and improve Bird's Eye View understanding across its autonomy stack.

United States Full-time Senior Machine Learning Engineer
$177k-$213k
C++ Computer Vision Deep Learning Machine Learning Python PyTorch
15 hours, 44 minutes ago
U

Applied AI Engineer

Unframe Inc. 51-200 Technology, Information and Internet

Unframe is hiring an Applied AI Engineer to work with enterprise customers and internal teams on deploying AI-driven solutions that connect business problems to production systems.

Israel Mid Level AI Engineer Machine Learning Engineer
Machine Learning Python React TypeScript
1 day, 14 hours ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers