Software Engineer - Training Infrastructure
Baseten · San Francisco · New York
CHECKED 58M AGO
Infrastructureremotesenior$165k–$330k
At a glance
Highlights
- Architect and lead development of training platform for frontier AI teams.
- Design a global training scheduler for large-scale ML workloads.
- Build reinforcement learning systems and continuous learning pipelines.
What they offer
- Meaningful equity (U.S. only)
- 100% coverage of medical, dental, and vision insurance 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)
Ideal candidate
Infrastructure-minded builder with 5+ years experience who loves owning large-scale distributed systems end to end and partnering with researchers to turn demanding ML training needs into reliable, high-performance platforms.
Requirements
- Bachelor's degree or higher in Computer Science or related field
- 5+ years of experience
- Proficiency in Go
- Deep expertise with Kubernetes in production environments
- Advanced understanding of distributed systems concepts and performance tuning
- Proven experience designing observability systems
- Experience with ML/AI workloads and MLOps platforms
Lab profile
About Baseten
Powers mission-critical AI inference to bring cutting-edge models into production.
What they build
- Mission-critical inference platform
- Flexible infrastructure for production models
- Developer tooling for AI deployment
Known for
- Applied AI research
- Flexible infrastructure
- Seamless developer tooling
Key facts
- Raised Series F