Baseten
Software Engineer - Training Infrastructure
$165K–$330K
San Francisco +1 · Hybrid · Full-time · Posted Aug 29, 2025
- Location
- San Francisco
New York - Experience
- 5+ years of experience.
- Education
- Bachelor’s degree or higher in Computer Science or related field.
About the job
Architect and develop infrastructure for Baseten’s ML training platform, enabling researchers and developers to run workloads with high performance and reliability.
What you'll do
- Design scalable training infrastructure, including scheduling, storage, networking, and a global training scheduler.
- Develop reinforcement learning systems and continuous learning pipelines, and improve system reliability and development velocity.
- Collaborate with research engineers, SRE, and Capacity teams; guide technical decisions and mentor junior engineers.
What they ask for
- Bachelor’s degree or higher in Computer Science or a related field.
- Proficiency in Go.
- Deep expertise with Kubernetes in production environments.
- Advanced distributed systems and performance-tuning knowledge; proven observability-system design experience; experience with ML/AI workloads and MLOps platforms.
Nice to have
- Experience with distributed storage systems; Python experience.
- Experience with major cloud providers, workload orchestration platforms, or open-source training frameworks and distributed training techniques.
Pay
$165K – $330K USD base salary
Tools
Go, Kubernetes, Python, AWS, GCP, Temporal, Airflow, PyTorch
AI focus
Build training infrastructure, reinforcement learning systems, and continuous learning pipelines for ML/AI workloads.
Model-written summary · the original posting is authoritative