Staff Product Manager, AI Platform

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

Staff Product Manager for Databricks' AI Platform, focusing on the infrastructure that powers machine learning and AI at scale. The role drives the vision and roadmap for AI platform product areas, enabling enterprises to build, train, deploy, and monitor AI/ML systems. It involves deep technical collaboration, customer engagement, and commercialization strategy for AI platform features.

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

  • 5+ years of experience as a Product Manager working on platform or infrastructure products
  • Deep technical background (CS, EE, or equivalent degree strongly preferred)
  • Experience with ML/AI infrastructure, data platforms, or cloud services
  • Proven enterprise B2B product management experience with highly technical customers

Nice to have

  • former software engineer experience is a significant plus
  • Familiarity with recommendation systems is a bonus

What the JD emphasized

  • Deep technical background
  • Experience with ML/AI infrastructure
  • Proven enterprise B2B product management experience

Other signals

  • AI Platform
  • ML lifecycle
  • production ML systems
  • LLM infrastructure
  • Agent infrastructure