Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Annapurna Labs builds custom Machine Learning accelerators that are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Neuron Compiler team is searching for compiler-skilled engineering talent to support the development and scaling of a compiler to enable the world's largest ML workloads to run performantly on these custom Annapurna systems.

As a Sr. Machine Learning Compiler Engineer III on the AWS Neuron Compiler team, you will be supporting the ground-up development and scaling of a compiler to handle the world's largest ML workloads. You will architect and implement business-critical features, publish cutting-edge research, and contribute to a brilliant team of experienced engineers. You will leverage your technical communications skill as a hands-on partner to AWS ML services teams and will be involved in pre-silicon design, bringing new products/features to market, and many other exciting projects.

Key responsibilities include:

• Solving challenging technical problems at every layer of the stack
• Designing, implementing, testing, deploying, and maintaining innovative software solutions
• Building high-quality, highly available, always-on products
• Researching implementations that deliver the best possible experiences for customers
• Participating in design discussions, code reviews, and communicating with internal and external stakeholders
• Working cross-functionally to help drive business decisions with technical input

The team values work-life harmony and offers flexibility in working arrangements. They provide mentorship, career growth opportunities, and a culture that celebrates knowledge-sharing and diversity.

A background in compiler development is strongly preferred, and experience in Machine Learning and AI accelerators is preferred but not required.