Software Engineer - Training Infrastructure

Job Description

Architect and lead the development of a scalable ML training platform, focusing on scheduling, storage, and networking. Partner with research engineers to translate complex training requirements into high-performance technical solutions.

Requirements: Requires a degree in Computer Science and deep expertise in Go and Kubernetes within production environments. Candidates must have a strong understanding of distributed systems and experience with ML/AI workloads.

Key Skills: Go, Kubernetes, Distributed Systems, Python, AWS, GCP, MLOps, Observability Systems, Infrastructure Architecture, Scheduling, Networking, PyTorch, DeepSpeed, FSDP, Temporal, Airflow

Benefits: Equity, Medical insurance, Dental insurance, Vision insurance, Flexible PTO, Winter Break, Paid parental leave, Fertility and family-building stipend, 401(k)