Software Development Engineer, Annapurna Labs, Trainium Collectives

Join Annapurna Labs, a crucial part of AWS, as a Software Development Engineer working on cutting-edge distributed AI/ML systems. This role focuses on developing collective operations that enable AI to scale across multiple accelerators and servers. You'll be working with a C/C++ stack and need strong knowledge of Linux, kernels, and performance optimization.

The position sits at the forefront of AI/ML development, where you'll build features for the largest clusters and AI models, working with major customers. As part of Annapurna Labs, you'll be contributing to hardware and software components that are fundamental to EC2 infrastructure - every EC2 instance runs on hardware designed by the team.

You'll work in a truly international environment alongside infrastructure experts, hardware engineers, RTL engineers, scientists, and architects. The team values mentorship, with opportunities to both receive guidance from senior engineers and mentor others. While the pace is fast-paced due to the rapidly evolving AI/ML field, the organization maintains a strong commitment to work-life balance and flexible working hours.

The role offers significant learning and growth opportunities, with access to principal-level engineers and directors. You'll be part of a team that celebrates knowledge-sharing and supports professional development. The position requires 3+ years of software development experience and strong system architecture skills.

Working at AWS means joining the world's most comprehensive cloud platform provider, where innovation is constant. You'll be part of an inclusive culture that values diverse experiences and perspectives. The company offers comprehensive benefits, mentorship opportunities, and a strong focus on work-life harmony. Whether you're an experienced developer or taking a non-traditional career path, AWS welcomes diverse backgrounds and experiences.