Sr. Software Development Engineer, HPC/ML Networking Engineer, Annapurna Labs

We are seeking an experienced engineer to work on distributed AI/ML systems. This role involves working on collective operations - the fundamental operations that enable AI to scale across multiple accelerators & servers. Most of our stack is C/C++ and relatively low level, so solid knowledge of Linux, kernels, and performant code is important. Experience with embedded systems is valued, and experience with high-speed networking or HPC interconnects is valued highly.

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 on the forefront of AI/ML, you’ll be working on features for the largest clusters, with the largest customers, for the largest AI models.

The org you would be joining is Annapurna Labs, an integral part of AWS and develops hardware and software components that are critical building blocks for EC2 infrastructure. Every instance in EC2 is running some type of hardware designed in Annapurna Labs. We specialize in designing software, systems and chips that optimize the AWS customer experience.

- Work on distributed AI/ML systems and collective operations

- Develop networking solutions for Machine Learning (ML) and High-Performance Computing (HPC) workloads on AWS

- Work side by side with infrastructure experts, hardware engineers, RTL engineers, scientists & architects

- Mentor new and junior engineers

- Design and architect new and existing systems with focus on reliability and scaling