Job Detail

Staff Software Engineer, Simulation ML Infrastructure

Data Science and AI Full–time
ID: #10602
Posted: 2026-03-19
Salary

Description

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. The Simulation ML Infrastructure team builds scalable AI/ML infrastructure to accelerate the Simulator team in sustainably innovating and building state of the art simulations of realistic environments for the testing and training of the Waymo Driver. To increase the fidelity and steerability of the simulations, we employ large foundation models trained on massive datasets to model the real world, including but not limited to, realistic agents (vehicles, pedestrians, cyclists, motorcyclists etc.), roads, traffic control systems, and weather etc. We seek an experienced Senior 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. You will: • Be part of a world-class, high-performing research engineering team to advance the state of the art of ultra realistic multi-agent simulations using foundation models. • Collaborate closely with the core Google DeepMind and Waymo Realism Modeling teams in London, and Waymo Oxford to use the large 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 deployment, and provide guidance on architectural decisions and technical directions. Own large, complex systems, driving architectures that meet technical and business objectives. • Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation and model training. • Collaborate cross-functionally to derive performance and system-level requirements for large ML systems. Translate product/business goals into measurable technical deliverables, ensuring system component alignment. • Mentor junior engineers, growing their expertise and fostering a collaborative culture. You have: • BS in Computer Science, Robotics, similar technical field of study, or equivalent practical experience • 5+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure such as developing, scaling, training, deploying, and optimizing large-scale machine learning systems from data to model. We prefer: • MS in Computer Science, Robotics, similar technical field of study, or equivalent practical experience • 10+ years of professional software engineering experience, with at least 5 years in machine learning infrastructure such as developing, designing, 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, or similar frameworks. • Strong expertise in distributed training techniques, including gradient sharding and optimization strategies for scaling large models across ML accelerator profiling tools to uncover performance bottlenecks. • Deep understanding of state-of-the-art machine learning models such as auto-regressive transformers and familiarity with custom-kernels for diverse h/w compute based efficiency. • Practical familiarity in Autonomous Driving, Simulations, and ML accelerators is a huge plus. 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 £150,000—£162,000 GBP

Hard Skills 10
Skill Source Confidence
Model Optimization llm_hard
100%
Neural Networks llm_hard
100%
Deep Learning llm_hard
100%
Transformers llm_hard
100%
Large Language Models (LLMs) llm_hard
100%
TensorFlow llm_hard
100%
PyTorch llm_hard
100%
Distributed Computing llm_hard
100%
Model Deployment llm_hard
100%
Gradient Descent llm_hard
80%
Soft Skills 6
Skill Source Confidence
Cross-Functional Communication llm_soft
100%
Leadership llm_soft
100%
Team Leadership llm_soft
100%
Mentoring llm_soft
100%
Collaboration llm_soft
100%
Problem-Solving llm_soft
100%
Apply Options
Publisher Direct Link
Glassdoor No Apply
Talents By StudySmarter No Apply
Glassdoor No Apply
API Logs for this Job
Query Country Status Response ms Created
Staff Software Engineer, Simulation ML Infrastructure extracted 8015 2026-03-22 02:31
Staff Software Engineer, Simulation ML Infrastructure classified 430 2026-03-21 21:02
junior machine learning engineer in London gb processed 9822 2026-03-21 16:57
Raw JSON
{
  "job_id": "5n80ZkNquo_mL4DdAAAAAA==",
  "job_city": "London",
  "job_state": null,
  "job_title": "Staff Software Engineer, Simulation ML Infrastructure",
  "job_country": "GB",
  "job_benefits": null,
  "job_latitude": 51.5072178,
  "job_location": "London",
  "job_onet_soc": "15111100",
  "apply_options": [
    {
      "is_direct": false,
      "publisher": "Glassdoor",
      "apply_link": "https://www.glassdoor.co.uk/job-listing/staff-software-engineer-simulation-ml-infrastructure-waymo-JV_IC2671300_KO0,52_KE53,58.htm?jl=1009644622430&utm_campaign=google_jobs_apply&utm_source=google_jobs_apply&utm_medium=organic"
    },
    {
      "is_direct": false,
      "publisher": "Talents By StudySmarter",
      "apply_link": "https://talents.studysmarter.co.uk/companies/jobflurry/staff-software-engineer-simulation-ml-infrastructure-7806754/?utm_campaign=google_jobs_apply&utm_source=google_jobs_apply&utm_medium=organic"
    },
    {
      "is_direct": null,
      "publisher": "Glassdoor",
      "apply_link": "https://www.glassdoor.co.uk/job-listing/staff-software-engineer-simulation-ml-infrastructure-waymo-JV_IC2671300_KO0,52_KE53,58.htm?jl=1009644622430"
    }
  ],
  "employer_logo": "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcSmyqTsPugxXK-0tLLZxEhEz5gtee9LTDLnDZEB&s=0",
  "employer_name": "Waymo",
  "job_is_remote": false,
  "job_longitude": -0.12758619999999998,
  "job_posted_at": "2 days ago",
  "job_publisher": "Glassdoor",
  "job_apply_link": "https://www.glassdoor.co.uk/job-listing/staff-software-engineer-simulation-ml-infrastructure-waymo-JV_IC2671300_KO0,52_KE53,58.htm?jl=1009644622430&utm_campaign=google_jobs_apply&utm_source=google_jobs_apply&utm_medium=organic",
  "job_highlights": {},
  "job_max_salary": null,
  "job_min_salary": null,
  "job_description": "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.\n\nThe Simulation ML Infrastructure team builds scalable AI/ML infrastructure to accelerate the Simulator team in sustainably innovating and building state of the art simulations of realistic environments for the testing and training of the Waymo Driver. To increase the fidelity and steerability of the simulations, we employ large foundation models trained on massive datasets to model the real world, including but not limited to, realistic agents (vehicles, pedestrians, cyclists, motorcyclists etc.), roads, traffic control systems, and weather etc.\n\nWe seek an experienced Senior 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.\n\nThis role reports to an Engineering Manager.\n\nYou will:\n• Be part of a world-class, high-performing research engineering team to advance the state of the art of ultra realistic multi-agent simulations using foundation models.\n• Collaborate closely with the core Google DeepMind and Waymo Realism Modeling teams in London, and Waymo Oxford to use the large models to improve sim realism.\n• 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 deployment, and provide guidance on architectural decisions and technical directions. Own large, complex systems, driving architectures that meet technical and business objectives.\n• Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation and model training.\n• Collaborate cross-functionally to derive performance and system-level requirements for large ML systems. Translate product/business goals into measurable technical deliverables, ensuring system component alignment.\n• Mentor junior engineers, growing their expertise and fostering a collaborative culture.\n\nYou have:\n• BS in Computer Science, Robotics, similar technical field of study, or equivalent practical experience\n• 5+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure such as developing, scaling, training, deploying, and optimizing large-scale machine learning systems from data to model.\n\nWe prefer:\n• MS in Computer Science, Robotics, similar technical field of study, or equivalent practical experience\n• 10+ years of professional software engineering experience, with at least 5 years in machine learning infrastructure such as developing, designing, scaling, training, deploying, and optimizing large-scale machine learning systems from data to model.\n• Solid experience in the development and optimization of machine learning infrastructure tools like DeepSpeed, PyTorch, TensorFlow, or similar frameworks.\n• Strong expertise in distributed training techniques, including gradient sharding and optimization strategies for scaling large models across ML accelerator profiling tools to uncover performance bottlenecks.\n• Deep understanding of state-of-the-art machine learning models such as auto-regressive transformers and familiarity with custom-kernels for diverse h/w compute based efficiency.\n• Practical familiarity in Autonomous Driving, Simulations, and ML accelerators is a huge plus.\n\nThe 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.\n\nSalary Range\n\n£150,000—£162,000 GBP",
  "job_google_link": "https://www.google.com/search?q=jobs&gl=gb&hl=en&udm=8#vhid=vt%3D20/docid%3D5n80ZkNquo_mL4DdAAAAAA%3D%3D&vssid=jobs-detail-viewer",
  "employer_website": "https://waymo.com",
  "job_onet_job_zone": "5",
  "job_salary_period": "YEAR",
  "job_apply_is_direct": false,
  "job_employment_type": "Full–time",
  "job_employment_types": [
    "FULLTIME"
  ],
  "job_posted_at_timestamp": 1773878400,
  "job_posted_at_datetime_utc": "2026-03-19T00:00:00.000Z"
}