Group Product Manager, ML Infrastructure

Waymo's Product Management Team is a mission-driven team responsible for writing and enacting first-of-their-kind playbooks to bring fully autonomous driving technology to market. Our team develops simple solutions for complex problems by orchestrating cross-functional efforts to safely drive Waymo's technology and the products it enables forward. We do this by understanding the customer, the business, and the technology. We are humble about the scope of our work, collaborative in our approach to problem-solving, and ambitious about our vision for the future.

You will:

• Define and maintain the multi-year product vision, strategy, and roadmap for Waymo's internal ML Infrastructure platform, encompassing data management, large-scale distributed training, rigorous evaluation, efficient deployment, and advanced MLOps capabilities
• Deeply understand the needs and workflows of Waymo's internal ML practitioners (engineers, researchers) through continuous engagement and collaboration, translating these needs into actionable platform improvements
• Partner closely with Engineering, Research, and other stakeholders to ensure alignment on platform priorities, define clear requirements, and drive the successful development and adoption of new capabilities
• Oversee the end-to-end product lifecycle, prioritizing initiatives that maximize ML developer velocity, model performance, system scalability, and efficient resource utilization
• Actively participate in technical discussions, offering guidance on ML frameworks, distributed systems design, hardware acceleration strategies, and MLOps best practices

You have:

• Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience
• 10+ years of combined experience in product management, technical program management, software engineering, or machine learning engineering
• 5+ years of direct experience building, launching, and iterating on technical platforms, cloud infrastructure, developer tools, or large-scale machine learning systems
• Demonstrated experience leading complex, technical projects with cross-functional teams from initial concept through successful delivery and adoption
• A strong understanding of the machine learning development lifecycle (data preparation, training, evaluation, deployment, monitoring)
• Familiarity with common ML frameworks (e.g., TensorFlow, PyTorch, JAX) and foundational infrastructure concepts including compute, storage, networking, and distributed systems

We prefer:

• Experience building internal developer platforms, tools, or infrastructure within a large, complex engineering organization
• Hands-on experience with technologies and challenges related to large-scale distributed training, MLOps implementation, ML model optimization, and leveraging hardware accelerators (GPUs, TPUs) effectively
• Proven ability to influence and align senior leadership (Director, VP level) and drive strategic initiatives in a dynamic, fast-paced, and technically demanding environment