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

Annapurna Labs, a crucial part of AWS, is seeking an experienced engineer to work on distributed AI/ML systems. This role focuses on developing collective operations that enable AI to scale across multiple accelerators & servers. The position involves working with C/C++ in a low-level environment, requiring solid knowledge of Linux, kernels, and performance optimization.

The team is at the forefront of AI/ML development, working on features for the largest clusters and AI models. You'll be part of a diverse, international workforce, collaborating with infrastructure experts, hardware engineers, RTL engineers, scientists & architects. The organization values mentorship, both receiving and providing guidance to team members.

Key Responsibilities:

• Developing and optimizing networking solutions for Machine Learning and High-Performance Computing workloads
• Working with HPC and ML customers to deliver scalable solutions
• Contributing to the development of fundamental operations for AI scaling
• Collaborating with cross-functional teams across hardware and software domains

The role offers:

• Flexible working hours and strong work-life balance
• Opportunity to work with principal-level engineers and directors
• Clear career growth paths and continuous learning opportunities
• International team environment with diverse perspectives
• Competitive compensation package ranging from $151,300 to $261,500 based on location

Required Qualifications:

• 5+ years of professional software development experience
• Strong programming skills and system architecture experience
• Experience with full software development lifecycle
• Leadership experience as a mentor or tech lead
• Deep understanding of Linux and kernel operations

The position is ideal for candidates passionate about solving complex problems in AI/ML infrastructure, with a focus on performance optimization and scalability. You'll be working in an environment that values innovation, mentorship, and professional growth while contributing to cutting-edge technology that powers AWS services.