Machine Learning Engineer, Waymo

• B.S. in CS, EE, Deep Learning or a related field
• 2+ years of industry experience
• Strong C++ programming skills
• Experience in developing on or using deep learning frameworks (e.g., PyTorch, TensorFlow, JAX, etc.)
• Passion for developing and optimizing ML software stacks for modern ML accelerator architectures (framework, runtime library, ML compiler, efficient deep learning etc.)

Desirable

• The ML Platform team at Waymo provides a set of tools to support and automate the lifecycle of the machine learning workflow, including feature and experiment management, model development, optimization and monitoring
• These efforts have resulted in making machine learning more accessible to teams at Waymo, including Perception, Planner, Research and Simulation
• We are looking for engineers with ML software & systems expertise to help us improve compute performance on both cloud and car
• You'll work across the entire ML stack from the system perspective, from efficient deep learning models, model compression, ML software (e.g. JAX, XLA, Triton, and CUDA), to
• You will be pleasantly challenged with deploying Waymo ML models on limited computation resources
• In this hybrid role, you will report to the Senior Manager of Runtime and Optimization
• Collaborate with the world-class Waymo ML practitioners in perception, planner, research and simulation to analyze ML workload performance and apply optimization techniques
• Assist in the development and implementation of efficient deep learning solutions
• Support the building of tools for benchmarking and optimizing deep learning models
• Contribute to the analysis of ML software stack and model performance
• Develop an efficient ML runtime system and optimized ML operator libraries for the diverse hardware options on Waymo’s platforms