Drimva

Lightning AI

Senior Product Manager, Experimentation Tooling

$160K–$275K

New York, New York, United States +3 · Manager · Full-time · Posted May 12, 2026

Location
New York, New York, United States
San Francisco, California, United States
New York, New York, United States
San Francisco, California, United States
Experience
7+ years in product management, including at least 3 years building infrastructure, platform, developer-tooling, or machine-learning products.
Education
BS in Computer Science, Engineering, or equivalent practical experience.

About the job

Owns the strategy, development, launch, and growth of Lightning AI’s experimentation and post-training product for AI researchers and engineers.

What you'll do

  • Set the product vision and roadmap for experimentation and post-training workflows, including integrations and product scope.
  • Work with engineering and customers to build and evaluate workflows for experiments, model quality, collaboration, and research-to-production handoffs.
  • Own pricing, packaging, metrics, and go-to-market activities, including technical customer and sales support.

What they ask for

  • 7+ years of product management experience, including at least 3 years building infrastructure, platform, developer-tooling, or machine-learning products.
  • Hands-on experience building products for ML engineers, AI researchers, or data scientists, with detailed understanding of experimentation and post-training workflows.
  • Experience with one or more post-training techniques and with designing or working closely with model evaluations.
  • BS in Computer Science, Engineering, or equivalent practical experience.

Nice to have

  • Experience at an AI infrastructure, neocloud, hyperscaler ML platform, experiment-tracking, or developer-tooling company.
  • Familiarity with PyTorch, PyTorch Lightning, distributed training, GPU infrastructure, large-scale fine-tuning, or listed ML tools.

Pay

$160,000 — $275,000 USD (base salary)

Tools

PyTorch, PyTorch Lightning, Weights & Biases, MLflow, Hugging Face, Ray, Slurm, Kubernetes

AI focus

Product work on AI experimentation and post-training, including fine-tuning, reinforcement learning, model evaluation, and workflows for selecting and moving models into production.

Model-written summary · the original posting is authoritative

First seen Oct 10, 2026 · Last verified Nov 5, 2026 (UTC)