companyWaymo LLC logo

Senior Software Engineer, Simulation Machine Learning Infrastructure

Waymo LLCLondon, UK
On-site Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Senior

Qualifications

Responsibilities:Join a high-performing research engineering team to push the boundaries of ultra-realistic multi-agent simulations utilizing foundation models. Work closely with the Google DeepMind and Waymo Realism Modeling teams in London and Oxford to leverage large models for enhanced simulation realism. Provide technical leadership in large-scale ML model architectures, particularly for autonomous vehicle models, guiding architectural decisions and ensuring alignment with technical and business objectives. Design and scale large distributed systems that encompass the entire ML lifecycle, enabling planet-scale dataset generation and model training. Collaborate across teams to establish performance and system-level requirements for large ML systems.

About the job

Waymo is at the forefront of autonomous driving technology, dedicated to becoming the world's most trusted driver. Originally initiated as the Google Self-Driving Car Project in 2009, our team has been relentlessly working to enhance mobility access and save lives by developing the Waymo Driver—The World’s Most Experienced Driver™. With over ten million successful rider-only trips facilitated by our technology, which has driven over 100 million miles on public roads and tens of billions of miles in simulations across more than 15 U. S. states, we are shaping the future of transportation.

The Simulation ML Infrastructure team is responsible for creating scalable AI/ML infrastructure that propels our Simulator team towards innovative, state-of-the-art simulations of realistic environments for the Waymo Driver's training and testing. By leveraging large foundation models trained on extensive datasets, we enhance the authenticity and control of our simulations, including realistic representations of vehicles, pedestrians, cyclists, roads, traffic control systems, and weather conditions.

We are currently seeking a highly skilled Senior Machine Learning Infrastructure Engineer to spearhead the development of advanced AI/ML infrastructure for multi-billion parameter foundation models tailored for ML accelerator-friendly simulations. Your extensive knowledge in massive model scaling, ML accelerators, and distributed training will be vital in designing and optimizing our systems.

About Waymo LLC

Waymo LLC is a pioneering technology company specializing in autonomous driving solutions. With a mission to revolutionize mobility across the globe, Waymo is committed to developing safe and efficient transportation systems that benefit communities everywhere.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.