Software Development Engineer, Annapurna Labs, Trainium Collectives

Join Annapurna Labs, a crucial part of AWS, in developing cutting-edge distributed AI/ML systems. As a Software Development Engineer, you'll work on collective operations that enable AI to scale across multiple accelerators and servers. The role involves working with C/C++ in a low-level environment, requiring solid knowledge of Linux, kernels, and performance optimization.

You'll be part of a diverse, international team working alongside infrastructure experts, hardware engineers, RTL engineers, scientists, and architects. The position offers exposure to the latest advancements in AI/ML, working with the largest clusters and customers on cutting-edge AI models.

The team values mentorship, both receiving and providing guidance to fellow engineers. You'll enjoy flexibility in working hours and a strong emphasis on work-life balance. Career growth opportunities are abundant, with access to principal-level engineers and directors.

Annapurna Labs specializes in designing software, systems, and chips that optimize the AWS customer experience. Every EC2 instance runs on hardware designed by the team, making your work crucial to AWS infrastructure. The fast-paced environment is balanced with team bonding and celebration of successes.

The role offers competitive compensation ranging from $129,300 to $223,600 based on location and experience, plus equity and comprehensive benefits. You'll be at the forefront of AI/ML development, working on features that impact the largest AI models and clusters while contributing to AWS's mission of being Earth's Best Employer.