CareerPlanGet AI match score →

Staff Product Manager, AI Platform

San Francisco, California💼 Full-time💰 $181,700–$181,700🗓 2026-04-09 → 2026-07-31

Core

Drive the vision and roadmap for AI platform areas to enable enterprises to build, train, deploy, and monitor production ML systems on Databricks.

Role type

Staff Product Manager, AI Platform Infrastructure

Builds

Unified, governed AI platform integrating MLflow, Unity Catalog, Model Serving, Vector Search, and LLM infrastructure for enterprise customers.

Domain

Enterprise Data & AI Infrastructure / Machine Learning Operations

Deliverable

production ML models

Required skills

Platform or infrastructure product management, ML/AI infrastructure knowledge, enterprise B2B sales experience, system architecture understanding, commercialization strategy, cross-functional leadership

Preferred skills

Former software engineer experience, familiarity with recommendation systems, deep technical background in CS/EE

Technologies

MLflow, Unity Catalog, Model Serving, Vector Search, LLM infrastructure, Apache Spark, Delta Lake

Responsibilities

Own product roadmap for AI platform areas, drive strategy for key AI platform capabilities, partner with engineering on technical ML infrastructure decisions, represent customer voice for enterprise ML teams, collaborate with GTM and Customer Success for adoption, define pricing and packaging strategy, grow end-user engagement by removing adoption bottlenecks

Seniority

Staff, hands-on IC with strategic scope

Rewrite
## Responsibilities - Drive the vision and roadmap for AI platform product areas and define how customers build, train, deploy, and monitor AI and ML systems on Databricks. - Collaborate across engineering teams to deliver an integrated and powerful path from experimentation to production. - Own the product roadmap for AI platform areas — defining what we build, why, and in what order — to accelerate customer adoption of AI and ML in production. - Drive strategy for key AI platform capabilities, shaping how enterprises operationalize AI at scale. - Partner closely with engineering teams to make deeply technical decisions about ML infrastructure — from distributed training architectures to real-time serving systems. - Represent the voice of the customer by engaging directly with enterprise ML teams, translating their pain points and workflows into platform capabilities that simplify the path to production AI. - Collaborate with GTM, Solutions Architecture, and Customer Success teams to drive enterprise adoption, shape field enablement, and inform competitive positioning. - Define pricing, packaging, and commercialization strategy for AI platform features, working with business teams to maximize value capture. - Grow end-user engagement with Databricks AI tools by identifying adoption bottlenecks and partnering cross-functionally to remove them. ## Requirements - 5+ years of experience as a Product Manager working on platform or infrastructure products, ideally in ML/AI, data, or cloud services. - Deep technical background — CS, EE, or equivalent degree strongly preferred; former software engineer experience is a significant plus. You should be comfortable going deep on system architecture, writing technical specs, and engaging credibly with world-class ML engineers. - Experience with ML/AI infrastructure, data platforms, or cloud services (e.g., model training, model serving, feature stores, vector search, LLM infrastructure, ML pipelines, or similar systems). - Proven enterprise B2B product management experience with highly technical customers — you have shipped platform products, driven commercial outcomes, and worked with field teams to land enterprise deals. ## Nice to Have - Familiarity with recommendation systems. ## Benefits - Work with a team that ships products used by thousands of the world's most sophisticated data and AI organizations. - Be part of a company that is passionate about enabling data teams to solve the world's toughest problems. - Collaborate across some of the most technically demanding areas in the platform, including recommendation systems, real-time inference, large-scale distributed training, LLM infrastructure, vector search, and feature stores. - Contribute to a mission to make it radically easier for enterprises to put AI into production by providing a unified, governed, and performant AI platform that integrates deeply with the Databricks Data Intelligence Platform.
Sourced via greenhouse · Listed on CareerPlan, which tracks 70,000+ jobs from 20+ sources.
Apply on Greenhouse ↗