Software Development Engineer, Annapurna Labs, Elastic Collectives at Annapurna Labs; U.S.) Inc

DESCRIPTION

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 and 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 highly prized.

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 that develops hardware and software components that are critical building blocks for EC2 infrastructure.

Every instance in EC2 runs some type of hardware designed in Annapurna Labs. We specialize in designing software, systems and chips that optimize the AWS customer experience.

In this position you’ll collaborate with infrastructure experts, hardware engineers, RTL engineers, scientists, and architects. Mentorship is a core part of our culture; you’ll receive senior guidance and be expected to mentor new and junior engineers. The pace is fast, but we also value team bonding and a healthy work‑life balance.

• 3+ years of non‑internship professional software development experience
• 2+ years of non‑internship design or architecture experience (design patterns, reliability and scaling) for new and existing systems
• Experience programming with at least one software programming language