ML Compiler Engineer, AWS Neuron, Annapurna Labs

Do you love decomposing problems to develop products that impact millions of people around the world? Would you enjoy identifying, defining, and building software solutions that revolutionize how businesses operate? The Annapurna Labs team at Amazon Web Services (AWS) is looking for a Software Development Engineer II to build, deliver, and maintain complex products that delight our customers and raise our performance bar. You'll design fault-tolerant systems that run at massive scale as we continue to innovate best-in-class services and applications in the AWS Cloud.

As a deep learning compiler engineer on the Neuron team, you will be a thought leader supporting the development of a compiler targeting AWS Inferentia and Trainum. You will be developing and scaling the compiler to handle the world's largest ML workloads. You will need to be technically capable, credible and curious in your own right as a trusted AWS Neuron engineer, innovating on behalf of our customers.

Key responsibilities:

• Solve challenging technical problems at every layer of the stack
• Design, implement, test, deploy and maintain innovative software solutions
• Build high-quality, highly available, always-on products
• Research implementations that deliver the best possible experiences for customers
• Create metrics, implement automation and other improvements
• Participate in design discussions, code review, and communicate with stakeholders
• Work cross-functionally to help drive business decisions with technical input

AWS values diverse experiences and encourages candidates from all backgrounds to apply. The team offers a good work-life balance, mentorship, and career growth opportunities.