Amazon
Lead Software Engineer, ML Network Stack - Annapurna Labs
$168K–$262K
Seattle, WA +1 · Full-time · Posted Jul 31, 2026
- Location
- Seattle, WA
Cupertino, CA - Experience
- 5+ years in systems design or architecture; 5+ years across the software development life cycle; 5+ years with C or C++; mentoring, tech leadership, engineering team leadership, or recent development experience within the last 3 years.
- Education
- Bachelor's degree in computer science or equivalent (preferred).
About the job
Leads development of the network stack and NVIDIA ML communication support that enables GPU-based AI/ML workloads to run on AWS EC2 systems.
What you'll do
- Lead engineers across the NVIDIA ML communication stack and direct work on features for large ML workloads.
- Develop and deliver support for ML frameworks, communication libraries, and EC2 networking.
- Guide senior, mid-level, and junior software development engineers.
What they ask for
- 5+ years leading design or architecture of new and existing systems, including design patterns, reliability, and scaling.
- 5+ years of full software development life cycle experience, including coding standards, reviews, source control, builds, testing, and operations.
- Experience as a mentor, tech lead, engineering team lead, or in development within the last 3 years.
- 5+ years of programming experience with C or C++.
Nice to have
- Bachelor's degree in computer science or equivalent.
- Experience with ML Communication Libraries or RDMA Networking.
Pay
USD 193,300.00 - 261,500.00 annually (base salary); USD 168,100.00 - 227,400.00 annually (base salary)
Tools
C, C++, NCCL, NVSHMEM, NIXL, NCCL GIN, CUDA kernels, Linux
AI focus
Leads network-stack and ML communication support for NVIDIA GPUs and large distributed AI/ML workloads on EC2.
Model-written summary · the original posting is authoritative