Software Engineer - Training Infrastructure

Baseten is seeking a Software Engineer for the Training Infrastructure team to architect and lead development of the training platform. Responsibilities include designing scalable infrastructure systems for ML training, partnering with research engineers, driving improvements in reliability and velocity, and mentoring junior engineers. The role involves critical architectural decisions and contributing to long-term technical strategy.

Bachelor's degree in CS or related, 5+ years experience, proficiency in Go, deep Kubernetes expertise, distributed systems knowledge, observability design, ML/AI and MLOps experience. Nice-to-have: Python, AWS/GCP, distributed storage, Temporal/Airflow, PyTorch, DeepSpeed, FSDP, NCCL.