SDE II, ML Infra Services, Annapurna Labs

Job Description

Lead the design and implementation of an ML infrastructure platform focusing on capacity management, workload scheduling, and fleet orchestration. Collaborate with ML scientists and hardware teams to ensure seamless accelerator access and high utilization.

Requirements: Requires 3+ years of professional software development and 2+ years of system design experience. Proficiency in Go, Java, or Python and experience with Kubernetes or container ecosystems is highly preferred.

Key Skills: Machine Learning Infrastructure, Capacity Management, Workload Scheduling, Fleet Orchestration, Profiling and Optimization, Resource Management, Code Generation, Kubernetes, Docker, Go, Java, Python, Javascript, Typescript, CI/CD, System Architecture

Benefits: Sign-on payments, Restricted stock units (RSUs), Health insurance, Medical insurance, Dental insurance, Vision insurance, Prescription insurance, Basic Life & AD&D insurance, Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage, 401(k) matching, Paid time off, Parental leave