Product Manager, Training Platform

Highlights

• Own the developer experience for training jobs end-to-end on Baseten.
• Turn distributed training complexity into product developers can depend on.
• Connect training with Loops, evaluation, and inference deployment.

What they offer

• Meaningful equity (U.S. only)
• 100% medical, dental, and vision coverage for employee and dependents
• Flexible PTO including company-wide Winter Break
• Paid parental leave
• Fertility and family-building stipend through Carrot (U.S. only)
• Company-facilitated 401(k)
• Exposure to a variety of ML startups for learning and networking

Ideal candidate

A builder with strong product taste and technical judgment who loves simplifying complex developer workflows. You partner closely with engineers and customers to turn emerging AI capabilities into adopted products.

Requirements

• Experience building and shipping AI/ML products or developer platforms
• A 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