200K–400K USD / year
Product Manager, Training Platform
ProductHybrid — San Francisco, New York
Published on 2026-11-11
These details were extracted automatically from the original listing. They may not be complete or fully up to date — it's worth checking the original job post.
About this role
Own the developer experience for running AI training and fine-tuning jobs on Baseten. Make distributed training workflows—including setup, scheduling, monitoring, debugging, and recovery—easier and more reliable for AI teams.
About the company
Baseten is an inference cloud that helps companies run and scale AI models in production with flexible infrastructure and developer tooling.
The team
EPD, Product
What you'll do
- Own the training-job experience from setup and submission through execution, monitoring, and recovery
- Simplify scheduling, GPU availability, logs, checkpointing, and debugging workflows
- Partner with research and infrastructure teams on supervised fine-tuning and reinforcement-learning workloads
- Help developers recover from failures, reproduce experiments, and spend less time waiting on infrastructure
- Connect training with Loops, model evaluation, and inference deployment
- Work directly with customers and turn their training needs into product improvements
What we're looking for
- 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 with teams that train, fine-tune, or adapt machine-learning models
- Understanding of distributed training, including GPU memory, scheduling, data access, checkpointing, and fault recovery
- Familiarity with infrastructure needs for supervised fine-tuning and reinforcement learning
- Strong instincts for developer workflows and simplifying complex technical systems
- Experience taking new capabilities from concept to production with researchers and engineers
Nice to have
- Hands-on experience running distributed training or building ML developer tools
- Experience with trainer and sampler systems for reinforcement learning
Benefits
- Meaningful equity
- Medical, dental, and vision insurance covered at 100% for U.S. employees and dependents
- Flexible PTO and company-wide winter break
- Paid parental leave
- Fertility and family-building stipend
- Company-facilitated 401(k) for U.S. employees