The Annapurna ML Software team is seeking a software engineer to develop mission-critical software for next-generation machine learning platforms. This role involves working at the intersection of hardware and software, focusing on the development of software and firmware for NeuronSwitches, which serve as the high-performance interconnect fabric for Trainium chips. Candidates will collaborate with various teams to ensure seamless integration and drive the qualification processes for new software implementations. This position is ideal for those passionate about low-level systems programming and hardware bring-up.

• 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
• 3+ years of full software development life cycle experience, including coding standards and testing
• Bachelor's degree in computer science or equivalent

• Architect and develop mission-critical software that powers Annapurna Labs' next-generation machine learning platforms
• Collaborate with EC2 teams and manufacturing partners to ensure seamless system integration
• Drive end-to-end qualification processes for new software implementations
• Craft high-performance solutions using C/C++ running on Linux
• Write device drivers and build ML infrastructure

• 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.