The role involves working with the Annapurna Labs team at AWS to develop high-performance kernels for machine learning operations on custom ML accelerators. Engineers will collaborate across various teams to optimize performance and contribute to future architecture designs. This position offers a unique opportunity to work at the intersection of machine learning, high-performance computing, and distributed architectures. The team values innovation and experimentation, fostering a culture of learning and collaboration.

• 3+ years of non-internship professional software development experience
• 2+ years of experience in design or architecture of new and existing systems
• Experience programming with at least one software programming language
• 3+ years of full software development life cycle experience
• Bachelor's degree in computer science or equivalent

• Design and implement high-performance compute kernels for ML operations
• Analyze and optimize kernel-level performance across multiple generations of Neuron hardware
• Conduct detailed performance analysis using profiling tools to identify and resolve bottlenecks
• Implement compiler optimizations such as fusion, sharding, tiling, and scheduling
• Work directly with customers to enable and optimize their ML models on AWS accelerators
• Collaborate across teams to develop innovative kernel optimization techniques

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