Product Manager, Training Platform

Skills

About the Role

You will own the developer experience for running training jobs, from environment setup and job submission to scheduling, debugging, checkpointing, and recovery. You'll work with engineers, researchers, and customers to make distributed training easier and connect training with evaluation and inference deployment.

Requirements

• Experience building and shipping AI/ML products or developer platforms
• Technical background in software engineering, ML engineering, research, or equivalent
• Direct experience working with teams that train, fine-tune, or adapt ML models
• Strong understanding of distributed training concepts including GPU memory, scheduling, data access, checkpointing, and fault recovery
• Familiarity with infrastructure needs of supervised fine-tuning and reinforcement learning
• Strong product instincts for developer workflows
• Experience partnering with researchers and engineers to take capabilities from concept to production

Responsibilities

• Own the product experience for training jobs from setup through execution, monitoring, and recovery
• Make distributed training easier with workflows for scheduling, GPU availability, logs, checkpoints, and debugging
• Work with research and infrastructure teams to support workloads including supervised fine-tuning and reinforcement learning
• Help developers recover from failures and reproduce experiments
• Connect training with Loops, model evaluation, and inference deployment
• Work with customers to understand how they train and adapt models and turn insights into product improvements

Benefits

• Equity
• 100% coverage of medical, dental, and vision insurance for employee and dependents (U.S. only)
• Flexible PTO policy including company-wide Winter Break
• Paid parental leave
• Fertility and family-building stipend through Carrot
• Company-facilitated 401(k) (U.S. only)