Job Opportunity: Director of Product Management, ML Frameworks and Generative AI Ecosystems at Google

Minimum Qualifications:

• Bachelor's degree or equivalent practical experience
• 15 years of product management experience with technical products

Preferred Qualifications:

• Experience in product managing internal Machine Learning (ML) infrastructure
• Advocating for its use cases and growing its adoption

About the Job

Google's mission is to drive the landing of machine learning (ML) powered products and innovation for all users by making ML easy to use. Our product managers encompass ML products for data management, experimentation, operations, training, and productionizing. We align our products with the latest research and product innovation, spanning traditional machine learning models and large models, including Large Language Models and Generative AI, across all Product Areas.

As the lead for ML Frameworks and Generative AI Ecosystems, you will be responsible for establishing and driving Google’s strategy for new and existing frameworks such as Jax, PyTorch, and TensorFlow for ML development. You will also oversee Google’s open source and community engagement for developer frameworks and tools to build a rich ecosystem that enables developers to incorporate the latest research capabilities into production quickly.

The generative AI space is rapidly evolving, presenting challenges in integrating generative AI across products to meet latency, cost, quality, reliability, and reproducibility requirements in high-impact industries. This phase of innovation requires deep customer empathy and forward-looking insights into where research innovation is headed. It also involves guiding customers through these transitions with well-supported tools and frameworks for both emerging and existing workloads.

You will stay abreast of industry developments while working closely with both research and product partners. Your role includes understanding the full end-to-end user journey for internal and external customers, identifying pain points, distilling user requirements and feedback, and mapping them into long-term strategic roadmaps. Identifying product features that add value and build trust is crucial.

Overview of Responsibilities

• Understand novel workflows for both creators and consumers of large models, identifying opportunities to create tools that empower generative-AI powered products.
• Partner deeply with Google DeepMind to understand researcher requirements for AI infrastructure, and create organizational processes, infrastructure roadmaps, developer solutions, and GTM strategies that accelerate developer workflows for production.
• Engage with both internal and external Foundational Model companies and researchers to grasp the user requirements and challenges of training state-of-the-art foundational models across TPUs and GPUs.
• Engage the open source community to identify opportunities to enrich the ecosystem and establish Google’s industry leadership in developer tooling and frameworks for generative AI.

Compensation and Benefits

The US base salary range for this full-time position is $253,000-$363,000, which includes bonus, equity, and benefits. Our salary ranges are determined by role, level, and location. The range displayed on each job posting reflects the minimum and maximum target for new hire salaries for the position across all US locations. Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant education or training. Your recruiter can provide more details about the specific salary range for your preferred location during the hiring process. Note that the compensation details listed in US role postings reflect the base salary only, with additional components such as bonuses, equity, and benefits.

Learn more about benefits at Google.

Company Information

Company Name: Google

Job Title: Director, Product Management, ML Frameworks and GenAI Ecosystems