companyWaymo LLC logo

Senior Software Engineer, Simulation ML Infrastructure

Waymo LLCMountain View, CA, USA; San Francisco, CA, USA
On-site Full-time $204K/yr - $259K/yr

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

Preferred Qualifications:Extensive experience in AI/ML infrastructure development. Proficiency in massive model scaling, ML accelerators, and large-scale distributed systems. Strong background in data engineering and model development. Ability to collaborate with cross-functional teams effectively.

About the job

Waymo is a pioneering company in autonomous driving technology, dedicated to becoming the world's most trusted driver. Originating from the Google Self-Driving Car Project in 2009, Waymo has committed to developing the Waymo Driver—The World’s Most Experienced Driver™—to enhance mobility access and save countless lives lost in traffic accidents. Our technology powers a fully autonomous ride-hail service and can be adapted to various vehicle platforms and applications. With over ten million rider-only trips and extensive experience driving more than 100 million miles on public roads and tens of billions of simulation miles across 15+ U. S. states, we are at the forefront of this transformative journey.

The Simulation ML Infrastructure team is focused on creating scalable AI/ML infrastructure that accelerates the Simulator team in developing state-of-the-art realistic simulations for testing and training the Waymo Driver. To enhance the realism and steerability of these simulations, we leverage large foundation models trained on vast datasets to accurately represent the real world, including realistic agents (vehicles, pedestrians, cyclists, motorcyclists), road systems, traffic control measures, and environmental conditions.

We are looking for a seasoned senior individual contributor to spearhead the advancement of sophisticated AI/ML infrastructure for multi-billion parameter foundation models within ML accelerator-friendly simulations. Your expertise in massive model scaling, ML accelerators, and large-scale distributed systems will be essential in designing and scaling our systems.

This position reports to an Engineering Manager.

Your Responsibilities:

  • Join a top-tier, high-performing research engineering team to push the boundaries of ultra-realistic multi-agent simulations using foundation models.
  • Collaborate closely with the Waymo Realism Modeling team located in London and the Waymo Oxford team to utilize large foundation models for enhancing simulation realism.
  • Operate at the intersection of data engineering, model development, and simulations, making architectural decisions. Take ownership of large, complex systems, and ensure that architectures and designs align with both technical and business objectives.
  • Design and scale extensive distributed systems that encompass the entire ML lifecycle, facilitating planet-scale dataset generation, model training, and evaluation.
  • Work cross-functionally to derive performance and system-level requirements for large ML systems. Convert product and business goals into measurable technical deliverables, ensuring alignment of system components.

About Waymo LLC

Waymo LLC is a leader in autonomous driving technology, driven by a mission to enhance mobility and safety on the roads. With its roots in the Google Self-Driving Car Project, Waymo has become synonymous with innovation in the field, delivering a fully autonomous ride-hail service and a wealth of experience in real-world driving and simulation.

Similar jobs

Tailoring 0 resumes

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