π Local Job Near You
Lead Engineer, ML Network Stack - Annapurna Labs
Amazon
π
Cupertino, United States
Location
Cupertino
Posted
June 02, 2026
Commute
Local Area
Local Opportunity Near You!
This job is in your area. Enjoy a short commute and work close to home.
Job Description
Description
We are seeking an experienced engineer and technical leader to join our team that owns the network stack for EC2 distributed AI/ML systems. The team develops support for a variety of frameworks and communication libraries including NCCL, NVSHMEM, NIXL, NCCL GIN, and Perplexity kernels. Solid knowledge of Linux, networking, and performant coding is important. Experience with embedded systems is valued, and experience with high-speed networking or HPC/RDMA interconnects is highly valued.
If you like solving hard problems, want to work with HPC and ML customers, iterate fast and deliver meaningful solutions at scale, then come join us! This truly is a role at the forefront of AI/MLβyou'll be working on features for the largest clusters, with the largest customers, for the largest AI models. This is a role for a technical lead with the expectation to grow into a technical manager role. We are specifically seeking candidates who want to develop their career as a tech...
We are seeking an experienced engineer and technical leader to join our team that owns the network stack for EC2 distributed AI/ML systems. The team develops support for a variety of frameworks and communication libraries including NCCL, NVSHMEM, NIXL, NCCL GIN, and Perplexity kernels. Solid knowledge of Linux, networking, and performant coding is important. Experience with embedded systems is valued, and experience with high-speed networking or HPC/RDMA interconnects is highly valued.
If you like solving hard problems, want to work with HPC and ML customers, iterate fast and deliver meaningful solutions at scale, then come join us! This truly is a role at the forefront of AI/MLβyou'll be working on features for the largest clusters, with the largest customers, for the largest AI models. This is a role for a technical lead with the expectation to grow into a technical manager role. We are specifically seeking candidates who want to develop their career as a tech...