Software Engineer (Training Product)

Join Baseten, a remote-first company focused on building developer-oriented tools for machine learning. As a Software Engineer, you will own features like multi-node training and serverless reinforcement learning, working through the stack from API to infrastructure. You will collaborate closely with research engineers and model developers, driving long-term improvements and resolving customer issues with urgency. Enjoy benefits like unlimited PTO, full healthcare coverage, and a learning and development budget.

• Conception to MVP: Own features like multi-node training and serverless reinforcement learning from conception to MVP.
• Architecting Solutions: Work through the stack, architecting solutions from API and UI down to our infrastructure layer.
• Collaboration: Work closely with research engineers leveraging state-of-the-art training techniques to build experiences that accelerate model development.
• Driven by high agency and ownership
• 5+ years experience building software applications
• Experience developing developer tooling or infrastructure products for external or internal users
• Strong communication skills with the ability to bridge technical depth and business needs
• Deep knowledge of the web stack, databases, and distributed systems
• Interest in ML/AI infrastructure and willingness to learn
• Good taste in product, particularly developer-oriented tools
• Experience with model development methods and paradigms, like Supervised Fine-Tuning, Reinforcement Learning, Synthetic Data Generation, LoRA, Full Finetunes, etc
• Experience launching features and products through different release cycles (MVP, Beta, GA, etc.)
• Familiarity or experience with the open source training stack and frameworks (NCCL, PyTorch, Megatron, NemoRL, VeRL, Axolotl, HF Trainer) and distributed training techniques (FSDP, DeepSpeed)
• Experience developing AI products, tooling, or agents
• Frontend fluency
• If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you