Senior Product Manager, ML Modeling & Platform

Klaviyo Klaviyo · Enterprise · Palo Alto, CA · Product

Senior Product Manager to own both the product strategy and roadmap for Klaviyo's core predictive and generative ML models (smart send time, audience optimization, product recommendations, churn prediction, AI-assisted content, agent capabilities) and the internal ML Platform infrastructure (training pipelines, experiment tracking, model serving, feature pipelines, orchestration tooling). The role requires deep understanding of ML systems, connecting platform investments to model outcomes and customer impact, and improving ML engineer developer experience at scale.

What you'd actually do

  1. Own the product strategy and roadmap for Klaviyo's core predictive and generative ML models — including smart send time, audience optimization, product recommendations, and churn prediction — defining what "better" looks like and how we get there
  2. Own the ML Platform roadmap: training infrastructure (DART/Ray), experiment tracking (MLflow), model serving, feature pipelines, and emerging tooling (Prefect) — making time-to-production for new models a first-class metric and driving it down continuously
  3. Build and maintain evaluation and monitoring frameworks so model quality is measurable, regressions are caught before they reach customers, and improvements compound over time
  4. Partner with engineering leadership on build vs. buy decisions across the ML stack — ensuring the platform evolves ahead of the needs of ML and AI product teams, not reactively behind them
  5. Connect platform and modeling investments to customer and business outcomes, working with go-to-market and customer success teams to ensure AI-powered features are well-understood and improving based on real feedback

Skills

Required

  • 5+ years of product management experience
  • Owning ML, AI, or data platform products in a production environment
  • Working knowledge of ML systems (training pipelines, model serving, experiment tracking, feature stores, or related infrastructure)
  • Understanding of tradeoffs in building and operating ML systems at scale
  • Ability to connect internal platform investments to customer and business outcomes
  • Ability to translate technical improvements into product stories
  • Comfort being the only PM in a highly technical room
  • Ability to drive decisions and defer to engineers
  • Systems thinking and understanding of tradeoffs

Nice to have

  • Experience with Ray, MLflow, Prefect
  • Experience with distributed training
  • Experience with LLM tooling

What the JD emphasized

  • ML Platform
  • ML models
  • production environment
  • ML systems
  • training pipelines
  • model serving
  • experiment tracking
  • feature stores
  • inference tradeoffs
  • ML developer experience
  • agentic products
  • generative AI
  • inference infrastructure
  • LLM tooling

Other signals

  • ML Platform
  • ML Models
  • Product Strategy
  • Infrastructure
  • Agentic Experiences