ML Network Stack Lead Engineer
Listed on 2026-10-06
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer
Annapurna Labs, an integral part of AWS, Cupertino, seeks a Lead Software Engineer to own the ML networking stack for EC2 distributed AI/ML systems. You will guide design across ML communication libraries (NCCL, NVSHMEM, NIXL) and CUDA kernels, mentoring engineers and shaping scalable software.
You will work on large-scale clusters, collaborate with hardware teams, and drive features for the largest AI models. Requires 5+ years leading design and C/C++ expertise.
We are seeking a motivated ML Network Stack Lead Engineer to join Amazon in Cupertino, CA, United States.
Full responsibilities and requirements are described in the listing above.
Learn more about the ML Network Stack Lead Engineer role in the description above.
We appreciate your interest in this position.
Join Amazon and contribute to our ongoing work.
Take a moment to read everything above and see whether this role is right for you.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).