Product Manager, AI Training Platform — Build Scalable ML

Baseten; San Francisco, New York (CA, NY); Hybrid; USD 180,000 - 240,000; Full time

Benefits offered by this job

Competitive compensation

Equity

100% medical/dental/vision for US

Flexible PTO including Winter Break

Parental leave

Fertility assistance

401(k)

Exposure to ML startups

Job summary

Baseten in San Francisco is seeking a Product Manager for Training Platform to own the developer experience for running training jobs, from environment setup through scheduling, debugging, and recovery. You’ll collaborate with engineers and researchers solving distributed training challenges and turn complexity into a reliable product for customers who train, fine-tune, and deploy models.

This role offers meaningful equity and competitive compensation as Baseten scales, with opportunities to

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 concepts, 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.

Responsibilities

• Own the product experience for training jobs, from setup and submission through execution, monitoring, and recovery.
• Make distributed training easier to use, with intuitive workflows for scheduling, GPU availability, logs, checkpoints, and debugging.
• Work closely with research and infrastructure teams to support different 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 the rest of Baseten's platform, including Loops, model evaluation, and inference deployment.
• Work directly with customers to understand how they train and adapt models, then turn those insights into product improvements.

Skills

AI/ML product experience

Developer platform

Distributed training knowledge

GPU memory understanding

Collaboration with researchers

Product instincts

Production collaboration

Job description

Baseten in San Francisco is seeking a Product Manager for Training Platform to own the developer experience for running training jobs, from environment setup through scheduling, debugging, and recovery. You’ll collaborate with engineers and researchers solving distributed training challenges and turn complexity into a reliable product for customers who train, fine-tune, and deploy models.

This role offers meaningful equity and competitive compensation as Baseten scales, with opportunities to