Senior ML Network Stack Engineer — Scale EC2 AI

Annapurna Labs, an integral part of AWS, seeks a Lead Software Engineer to own the ML network stack for EC2 distributed AI/ML systems. You will lead across the ML communication stack, mentor engineers, and drive scalable features for the largest AI workloads. You'll work with NCCL, NVSHMEM, NIXL, NCCL GIN and CUDA kernels, while applying Linux networking and HPC-style interconnects to optimize EC2 infrastructure for ML customers.