Baseten
Software Engineer - Training Product
$165K–$330K
San Francisco +1 · Hybrid · Full-time · Posted Jan 22, 2026
- Location
- San Francisco
New York - Experience
- 5+ years experience building software applications.
About the job
Build and launch training products for AI models, working across APIs, backend, databases, and infrastructure to improve customer training workflows.
What you'll do
- Design APIs and abstractions for complex resources and lifecycles, and implement features across the stack.
- Fine-tune and deploy models; collaborate with model developers and research engineers on post-training workflows.
- Improve system reliability and development velocity, and resolve bugs and customer issues.
What they ask for
- 5+ years experience building software applications.
- Deep knowledge of the web stack, databases, and distributed systems.
- Experience developing developer tooling or infrastructure products for external or internal users.
- Interest in ML/AI infrastructure and willingness to learn.
Nice to have
- Experience launching features and products through release cycles such as MVP, Beta, and GA.
- Experience with model development methods, AI products or tooling, open source training frameworks, distributed training, or frontend development.
Pay
$165K – $330K USD base salary
Tools
Kubernetes, vLLM, NCCL, PyTorch, Megatron, NemoRL, VeRL, Axolotl
AI focus
Build AI model-training products, fine-tune and deploy models, and support post-training workflows including reinforcement learning.
Model-written summary · the original posting is authoritative