Machine Learning Engineer, Runtime & Optimization

Who you are

You have a strong background in machine learning software or systems, with experience in optimizing compute performance for both cloud and edge devices. Your expertise includes deploying efficient deep learning models and model compression techniques, ensuring that machine learning workflows are streamlined and effective. You are familiar with tools such as JAX, TensorFlow, and CUDA, and you understand the intricacies of working with limited computational resources. You thrive in collaborative environments, where you can lead efforts to improve the ML stack and contribute to the overall mission of enhancing autonomous driving technology. You are detail-oriented and have a passion for making machine learning accessible to various teams within the organization.

Desirable

Experience with ML platforms and tools that support the lifecycle of machine learning workflows is a plus. Familiarity with performance optimization techniques and a strong understanding of the challenges associated with deploying models in real-world scenarios will set you apart. You are eager to tackle complex problems and are motivated by the impact of your work on improving mobility and safety.

What you'll do

In this role, you will lead the collaboration with cross-functional teams to enhance the performance of Waymo's machine learning models. You will work across the entire ML stack, focusing on optimizing model deployment and ensuring that the models run efficiently on both cloud infrastructure and vehicle platforms. Your responsibilities will include developing and implementing strategies for model compression and performance tuning, as well as monitoring and evaluating the effectiveness of deployed models. You will also contribute to the development of tools that automate the machine learning workflow, making it easier for teams to manage features and experiments.

You will engage with various teams, including Perception, Planner, Research, and Simulation, to understand their needs and provide tailored solutions that enhance their machine learning capabilities. Your role will involve hands-on coding and experimentation, allowing you to stay at the forefront of machine learning technology while driving impactful results.

What we offer

Waymo provides a competitive salary range of $204,000—$259,000 USD, along with eligibility for an annual bonus program and equity incentive plan. You will be part of a mission-driven company focused on improving access to mobility and saving lives through autonomous driving technology. The work environment is collaborative and innovative, with opportunities for professional growth and development. We encourage you to apply even if your experience doesn't match every requirement, as we value diverse perspectives and backgrounds.