jobdescription_image_2x.png

Staff Machine Learning Infrastructure Engineer, Simulation

London, England, United Kingdom Full-Time Software Engineering 3368

Apply now

Staff Machine Learning Infrastructure Engineer, Simulation

  • 3368
  • On Site
  • London, England, United Kingdom
  • Software Engineering
  • Full-Time
  • Advanced Career

Waymo is an autonomous driving technology company with the mission to be the 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 One, a fully autonomous ride-hailing service, and can also be applied to a range of vehicle platforms and product use cases. The Waymo Driver has provided over one million rider-only trips, enabled by its experience autonomously driving tens of millions of miles on public roads and tens of billions in simulation across 13+ U.S. states.

The Simulation Machine Learning 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 realistic agents (vehicles, pedestrians, cyclists, motorcyclists), roads, traffic control systems, and weather

We are looking for an experienced Staff 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.

In this role, you’ll:

  • Be part of a world-class, research engineering team to improve the of ultra realistic multi-agent simulations using foundation models.
  • Collaborate with the core Waymo Realism Modeling team in London and Waymo Oxford to use large foundation 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 simulations, and provide guidance on architectural decisions and technical directions. Manage complex systems, driving architectures that meet technical and goals.
  • Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation, model training, and evaluation.
  • Collaborate to derive performance and system-level requirements for large ML systems. Translate product/business goals into measurable technical deliverables, ensuring system component agreement.
  • Mentor junior engineers, growing their expertise and promoting a collaborative culture.

Preferred qualifications

  • 8+ years of professional software engineering experience, with at least 5 years in machine learning infrastructure such as developing, 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, Ray, or similar frameworks.
  • Expertise in distributed training techniques, including gradient sharding and optimization strategies for scaling large models across ML accelerator profiling tools to uncover performance bottlenecks. Familiarity with custom kernels for compute based efficiency.
  • Experience with state-of-the-art machine learning models such as autoregressive transformers. 
  • With experience navigating cross-functional teams and providing technical leadership projects across multiple organizations.
  • , with the ability to translate complex technical concepts for a broad audience.
  • Practical familiarity in Autonomous Driving, Simulations, and ML accelerators is a 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
£145,000—£157,000 GBP

We appreciate your interest in Waymo. Waymo is an equal employment opportunity employer, committed to maintaining a supportive and inclusive workplace for all employees. Waymo does not discriminate against, and prohibits harassment of, any applicant or employee based on race, color, sex, sexual orientation, gender identity, religion, national origin, age, disability, military status, genetic information or any other basis protected by applicable law. Waymo will also consider for employment qualified applicants with criminal records in accordance with applicable law. Waymo is committed to ensuring equal opportunity for qualified individuals with disabilities. If you are an individual with a disability and require an accommodation to participate in the application or interview process, please let the recruiting team know or email waymo-candidatesupport@google.com. (This email address is intended to be used only for requesting accommodations as part of the application process. Other inquiries will not receive a response.)

Some of our Benefits

Health and wellness

Our people are at the heart of everything we do. At Waymo, you can enjoy top-notch medical, dental and vision insurance, mental wellness support, gym membership, and special wellness programs.

Financial wellness

Your financial peace of mind is important to us. At Waymo, we offer competitive compensation, bonus opportunities, equity, employees provident fund, and lots of other perks and employee discounts.

Flexibility and time off

Take the time you need to relax and recharge. Enjoy the flexibility to work from another location for four weeks per year. We support an on-site or hybrid work model and offer remote working opportunities, paid time off, bereavement, sick, and parental leave. 

Supporting families

When it comes to growing your family or caring for your loved ones, you have our full support. Enhanced leave options include 26 weeks of paid leave for birthing parents and 18 weeks of paid leave for non-birthing parents.

Community and personal development

At Waymo, you'll find a range of opportunities to grow, connect, and give back. We offer education reimbursement, personal and professional development, mentorship, and other ways to connect through Employee Resource Groups (ERGs), other internal groups, and even time off to volunteer.

Cool perks

Access to Google offices, cafes, wellness centers, massages, and so much more. To support your wellbeing at home, you can enjoy at-home fitness and cooking classes, and more.

Ready to Apply?

 

To ensure thorough consideration of each application, candidates may submit up to 5 job applications within a 60-day period. We encourage you to carefully review job descriptions and apply to positions that best match your skills. Applications beyond this limit will not be considered.

Check this box to join the talent community and sign up for job alerts

Not You?

We have emailed you a code to verify your identity. Please check your spam/junk folder if you don't receive the email in your inbox.

Thank you