Product Manager, Training Platform
Core
Own the developer experience for running training jobs on Baseten, from environment setup and job submission to scheduling, debugging, checkpointing, and recovery.
Role type
Product Manager, Training Platform
Builds
A training platform for AI teams to focus on improving models instead of managing infrastructure
Domain
AI/ML Infrastructure, Distributed Training
Deliverable
production ML models
Required skills
AI/ML product building, distributed training concepts, GPU memory management, scheduling, data access, checkpointing, fault recovery, supervised fine-tuning, reinforcement learning, developer workflow simplification
Preferred skills
Hands-on distributed training, ML developer tools, trainer and sampler systems for reinforcement learning
Technologies
GPU, distributed training, supervised fine-tuning, reinforcement learning
Responsibilities
Own the product experience for training jobs, Make distributed training easier to use, Work closely with research and infrastructure teams, Help developers recover from failures, Connect training with the rest of Baseten's platform, Work directly with customers to understand training needs
Seniority
Mid-Senior, hands-on IC
