We are seeking an experienced engineer to work on distributed AI/ML systems, focusing on collective operations that enable AI to scale across multiple accelerators and servers. The role is part of Annapurna Labs, a crucial part of AWS, which develops hardware and software components for EC2 infrastructure. You will work alongside a diverse team of experts in a fast-paced environment, emphasizing mentorship and career growth. The position offers flexibility in working hours and values work-life balance while tackling challenging problems in the AI/ML field.

• 3+ years of non-internship professional software development experience
• 2+ years of non-internship design or architecture experience of new and existing systems
• Experience programming with at least one software programming language
• Knowledge of Linux fundamentals
• 3+ years of full software development life cycle experience, including coding standards and code reviews
• Bachelor's degree in computer science or equivalent
• Knowledge of Python and/or C++ programming

• Work on collective operations that enable AI to scale across multiple accelerators and servers
• Collaborate with infrastructure experts, hardware engineers, and scientists
• Mentor new and junior engineers while receiving mentorship from senior members
• Develop networking solutions for Machine Learning (ML) and High-Performance Computing (HPC) workloads
• Participate in the full software development life cycle, including coding, testing, and operations

• Employees at Amazon are often offered comprehensive health benefits—including multiple medical plan options (no pre-existing condition exclusions, 100% covered in-network preventive care), dental and vision plans, a 24/7 medical advice line from day one, expert second-opinion services, and broad mental-health support with several free counseling sessions (including pediatric). Financial wellness typically includes a 401(k) with company match (up to 2%), Restricted Stock Units (equity), FSAs, an emergency savings program, product and partner discounts, and even college-savings and home-purchase programs. Overall, the package is designed to support employees and their families’ health, finances, and day-to-day life.