[Remote] Staff Product Manager, AI Platform

Note: The job is a remote job and is open to candidates in USA. Databricks is a data and AI company that empowers data teams to tackle complex challenges through its advanced data and AI infrastructure platform. The Staff Product Manager for the AI Platform will lead the vision and roadmap for AI product areas, collaborating with engineering teams to enhance customer engagement and operationalize AI at scale.

Responsibilities

• 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
• You will drive strategy for key AI platform capabilities, shaping how enterprises operationalize AI at scale
• You will partner closely with engineering teams to make deeply technical decisions about ML infrastructure — from distributed training architectures to real-time serving systems
• 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
• You will collaborate with GTM, Solutions Architecture, and Customer Success teams to drive enterprise adoption, shape field enablement, and inform competitive positioning
• You will define pricing, packaging, and commercialization strategy for AI platform features, working with business teams to maximize value capture
• You will grow end-user engagement with Databricks AI tools by identifying adoption bottlenecks and partnering cross-functionally to remove them

Skills

• 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). Familiarity with recommendation systems is a bonus
• 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

Company Overview

Databricks is a data and AI platform that unifies data engineering, analytics, and machine learning on a lakehouse architecture. It was founded in 2013, and is headquartered in San Francisco, California, USA, with a workforce of 5001-10000 employees. Its website is https://www.databricks.com.

Company H1B Sponsorship

Databricks has a track record of offering H1B sponsorships, with 174 in 2026, 385 in 2025, 319 in 2024, 227 in 2023, 222 in 2022, 166 in 2021, 64 in 2020. Please note that this does not guarantee sponsorship for this specific role.