CareerPlanGet AI match score →

Senior Product Manager, Experimentation Tooling

New York, New York💼 Full-time💰 $160,000–$160,000🗓 2026-06-21 → 2026-07-31

Core

Own Lightning AI's experimentation and post-training product end-to-end, enabling AI researchers and engineers to turn ideas into validated models with a cohesive workflow.

Role type

Senior Product Manager (AI Infrastructure/Developer Tooling)

Builds

Experimentation and post-training platform features for model development, evaluation, and production deployment.

Domain

Artificial Intelligence / Machine Learning Infrastructure

Deliverable

product features

Required skills

Product strategy and roadmap ownership, ML engineer and AI researcher workflow expertise, post-training techniques (fine-tuning, RL, distributed training), model evaluation design, pricing and packaging strategy, technical depth in APIs/SDKs/observability, end-to-end product ownership, prioritization and scope management.

Preferred skills

Experience at AI infrastructure/neocloud/hyperscaler, familiarity with PyTorch/PyTorch Lightning, experience with W&B/MLflow/Hugging Face/Ray/Kubernetes, building collaborative research-to-production workflows.

Technologies

PyTorch, PyTorch Lightning, W&B, MLflow, Hugging Face, Ray, Slurm, Kubernetes

Responsibilities

Define product vision and roadmap for post-training and experimentation; build products that make ML workflows coherent; work directly with engineers from problem definition through launch; own model evaluation as a product function; design pricing and packaging; build collaborative workflows for research-to-production; serve as product voice in GTM.

Seniority

Senior, hands-on IC

Sourced via greenhouse · Listed on CareerPlan, which tracks 70,000+ jobs from 20+ sources.
Apply on Greenhouse ↗