Staff Product Manager, AI Platform

Databricks Databricks · Data AI · San Francisco, CA · Product

Staff Product Manager for Databricks' AI Platform, focusing on driving the vision and roadmap for AI/ML infrastructure products that enable enterprises to build, train, deploy, and monitor AI systems. The role involves deep technical collaboration with engineering, customer engagement, and commercialization strategy for AI platform features, aiming to make it easier for enterprises to put AI into production.

What you'd actually do

  1. You will 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.
  2. You will drive strategy for key AI platform capabilities, shaping how enterprises operationalize AI at scale.
  3. You will partner closely with engineering teams to make deeply technical decisions about ML infrastructure — from distributed training architectures to real-time serving systems.
  4. You will 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.
  5. You will collaborate with GTM, Solutions Architecture, and Customer Success teams to drive enterprise adoption, shape field enablement, and inform competitive positioning.

Skills

Required

  • Product Management
  • Platform/Infrastructure Product Experience
  • ML/AI Infrastructure
  • Data Platforms
  • Cloud Services
  • Enterprise B2B Product Management
  • Technical Background (CS/EE preferred)
  • Customer Engagement
  • Roadmap Definition
  • Strategy Development

Nice to have

  • Recommendation Systems
  • Former Software Engineer Experience

What the JD emphasized

  • 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.
  • 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

Other signals

  • drive the vision and roadmap for AI platform areas
  • define how customers build, train, deploy, and monitor AI and ML systems
  • partner closely with engineering teams to make deeply technical decisions about ML infrastructure
  • represent the voice of the customer by engaging directly with enterprise ML teams
  • define pricing, packaging, and commercialization strategy for AI platform features