Senior SWE, Inference, Machine Learning Infrastructure

The Waymo ML Infrastructure team works with Research and Production teams to develop state of the art models in the areas of Perception and Planning that are core to our autonomous driving software. We enable our partners by offering the best in class solutions for the entire model development lifecycle. These solutions are developed in close collaboration with teams at Google. They are geared towards both scaling models and solving problems unique to ML for autonomous driving.

You will

In this role you will focus on developing large-scale and efficient inference solutions for Waymo. This is crucial to the development of Waymo Foundation Models for driving including evaluation, auto-labeling and distillation. We are looking for an individual contributor (IC) who enjoys the following responsibilities, and has the required qualifications listed below:

You Have

• B.S. in Computer Science, Math, or equivalent real-world experience
• Solid understanding of machine learning fundamentals and experience with popular ML frameworks such as JAX, PyTorch, or TensorFlow 
• Experience with building and optimizing scalable inference platforms for real-time and batch processing.
• Practical familiarity using profiling tools to uncover performance bottlenecks
• Experienced in software development projects using C++ or Python and deploying on cloud platforms, for example AWS, Azure, or GCP.

We Prefer

• MS or PhD in Computer Science, Math
• Experience with distributed system and multi-threaded and stream-based programming models
• Practical familiarity using ML accelerator profiling tools to uncover performance bottlenecks

This role reports to an Engineering Manger

#LI-Hybrid