Tech Lead Manager (ML Optimization)

Join Waymo, the leading autonomous driving technology company, as a Tech Lead Manager (ML Optimization). In this critical role, you will lead the development and deployment of large-scale machine learning models, working cross-functionally at the intersection of data engineering, model development, and low-latency deployments. You will drive model efficiency, guide efforts across multiple teams, and mentor junior engineers. Enjoy a comprehensive benefits package, including medical, dental, and vision insurance, competitive compensation, and a hybrid work model.

• Lead the development and enable efficient deployment for large-scale machine learning models using state-of-the-art advanced AI infrastructure.
• Take ownership of improving model efficiency on different platforms and drive the model system codesign practice that meets both technical and business requirements.
• Guide efforts across multiple teams and organizations to ensure seamless integration of data generation, model development, and deployment pipelines.
• Deep understanding of state-of-the-art machine learning models and architectures such as autoregressive and diffusion transformers and familiarity with custom-kernels for diverse h/w compute based efficiency
• A Master’s or PhD in Computer Science, Engineering, or a related field is preferred
• Strong leadership skills with experience navigating cross-functional teams and providing technical leadership projects across multiple organizations
• Solid experience in the development and optimization of machine learning infrastructure tools like DeepSpeed, PyTorch, TensorFlow, JAX, or similar frameworks
• Excellent communication skills, both verbal and written, with the ability to translate complex technical concepts for a broad audience
• 10+ years of professional software engineering experience, with at least 5 years in machine learning infrastructure such as developing, training, deploying, and optimizing large-scale machine learning systems
• Experienced using ML accelerator profiling tools to uncover performance bottlenecks