Product Manager, Training Platform

The role

You will own the product experience for running training jobs on Baseten, from setup and submission through scheduling, debugging, checkpointing and recovery. You will work with engineers, researchers and customers to make model training simpler for AI teams.

What you will do

• Own the product experience for training jobs, from setup and submission through execution, monitoring and recovery
• Make distributed training easier to use, with workflows for scheduling, GPU availability, logs, checkpoints and debugging
• Work with research and infrastructure teams on training workloads, including supervised fine-tuning and reinforcement learning
• Help developers recover from failures, reproduce experiments and spend less time waiting on infrastructure
• Connect training with Baseten's platform, including Loops, model evaluation and inference deployment
• Work directly with customers and turn their training insights into product improvements

Required qualifications

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

Nice to have

• Hands-on experience running distributed training, building ML developer tools, or working with trainer and sampler systems for reinforcement learning

Benefits

• Competitive compensation, including meaningful equity
• U.S. only: 100% coverage of medical, dental and vision insurance for employee and dependents
• Flexible PTO policy, including company-wide Winter Break (offices closed from Christmas Eve to New Year's Day)
• Paid parental leave
• Fertility and family-building stipend through Carrot
• U.S. only: Company-facilitated 401(k)
• Exposure to a variety of ML startups, with learning and networking opportunities

Before you apply

• The posting is an Equal Opportunity Employer and will consider qualified applicants with criminal histories in line with applicable law, including the San Francisco Fair Chance Ordinance where applica

About the company

Baseten powers inference for AI companies such as Cursor, Notion and Clay, and is building a cloud where inference, training and agent execution share one platform. The company recently raised a $1.5B Series F led by Altimeter Capital, Conviction Partners and Spark Capital.