Sr ML Compiler Engineer, Annapurna Labs

AWS Machine Learning accelerators are leading AWS innovation in Generative AI development. The role focuses on the Inferentia chip and Trainium platforms, which deliver best-in-class ML inference and training performance in the cloud. As a Sr. Machine Learning Compiler Engineer III on the AWS Neuron team, you'll lead the development of a compiler handling the world's largest ML workloads. The position involves working with the AWS Neuron SDK, which includes an ML compiler, runtime, and integration with frameworks like PyTorch, TensorFlow, and MxNet.

You'll be part of the Amazon Annapurna Labs team, responsible for silicon development at AWS. The role requires architecting and implementing critical features, publishing research, and mentoring experienced engineers. You'll collaborate with AWS ML services teams and participate in pre-silicon design and new product launches.

The position offers comprehensive benefits, including medical, financial, and other benefits. AWS values work-life harmony and promotes an inclusive culture through employee-led affinity groups and inclusion events. The company provides extensive opportunities for knowledge-sharing, mentorship, and career advancement.

Key customers using AWS Neuron and Inferentia include Snap, Autodesk, Amazon Alexa, and Amazon Rekognition. The role requires 5+ years of professional software development experience, including leadership in design/architecture and full software development lifecycle expertise.

The compensation package is competitive, ranging from $151,300 to $261,500 per year based on geographic location, plus equity, sign-on payments, and comprehensive benefits. AWS values diverse experiences and welcomes candidates from non-traditional backgrounds.