companyWaymo LLC logo

Machine Learning Engineer - Runtime & Optimization

Waymo LLCMountain View, California, USA
Hybrid 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

Mid to Senior

Qualifications

You will:Collaborate with world-class Waymo ML scientists across perception, planning, research, and simulation to identify system and model optimization opportunities for faster ML workloads. Lead project proposals through execution while mentoring junior engineers. Analyze and enhance ML system workloads on both cloud infrastructures and self-driving vehicles. Implement model optimization and efficient deep learning techniques to bolster Waymo’s ML systems. You have:A Master’s degree in Computer Science, Electrical Engineering, Deep Learning, or a related field. At least 2 years of experience as a technical lead, which includes creating project plans, engaging with customer teams, mentoring, and being accountable for achieving goals and execution. Over 5 years of experience in developing solutions within ML systems or the ML software stack (Pytorch/JAX/TF, runtime libraries, ML compilers). A profound understanding of ML system architecture, performance analysis, and relevant tools. Strong programming skills in Python or C++.

About the job

Waymo is at the forefront of autonomous driving technology, dedicated to becoming the world’s most trusted driver. Originating from the Google Self-Driving Car Project in 2009, Waymo has been relentlessly focused on creating the Waymo Driver—The World’s Most Experienced Driver™—to enhance mobility access and prevent traffic-related fatalities. The Waymo Driver is the driving force behind our fully autonomous ride-hail service, optimized for various vehicle platforms and applications. With over ten million rider-only trips and more than 100 million miles driven autonomously on public roads, alongside tens of billions of simulated miles across 15+ U. S. states, Waymo is redefining transportation.

The ML Platform team at Waymo plays a critical role by offering a suite of tools that facilitate and automate the entire machine learning workflow lifecycle, including feature and experiment management, model development, optimization, and monitoring. Our initiatives have made machine learning more accessible across diverse teams at Waymo, including Perception, Planner, Research, and Simulation.

We are seeking talented engineers with expertise in machine learning software or systems to enhance compute performance both in the cloud and on vehicles. You'll engage with the entire ML stack from a systems perspective, tackling challenges such as efficient deep learning models, model compression, and improving ML software (e.g., JAX, XLA, Triton, and CUDA). This hybrid position reports directly to the Senior Manager of Runtime and Optimization.

About Waymo LLC

Waymo LLC is a pioneering company in the realm of autonomous driving technology, aiming to make transportation safer and more accessible through innovative solutions that reduce traffic fatalities. With a robust background stemming from Google, Waymo continues to lead the industry by developing the most experienced autonomous driving systems.

Similar jobs

Tailoring 0 resumes

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