Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Annapurna Labs, an AWS subsidiary, is seeking a Senior Compiler Engineer III to join their Neuron Compiler team. This role focuses on developing and scaling compilers for AWS's custom Machine Learning accelerators, specifically the Inferentia/Trainium chips. These accelerators are crucial for AWS's AI capabilities and are used by major customers like Snap, Autodesk, and Amazon Alexa.

The position involves working on a deep learning compiler stack that optimizes LLM and Vision models from frameworks like TensorFlow, PyTorch, and JAX for AWS's custom accelerators. You'll be part of a team of experienced engineers working on cutting-edge technology that powers some of the world's largest ML workloads.

As a Sr. Machine Learning Compiler Engineer III, you'll contribute to ground-up development, architect critical features, publish research, and collaborate with AWS ML services teams. The role requires strong technical communication skills and offers involvement in pre-silicon design and new product development. While compiler development experience is strongly preferred, ML and AI accelerator background is beneficial but not mandatory.

The team operates in a startup-like environment within AWS, focusing on high-impact solutions and innovation. They value knowledge-sharing, mentorship, and foster an inclusive culture through various employee-led affinity groups and ongoing learning experiences. The position offers work-life harmony through flexible hybrid work arrangements, allowing for both focused individual work and in-person collaboration.

AWS provides comprehensive compensation including competitive base pay (ranging from $151,300 to $261,500 based on location), equity, sign-on payments, and extensive benefits. The role offers significant growth opportunities within AWS's expanding ML infrastructure, making it an excellent opportunity for experienced compiler engineers looking to impact the future of machine learning acceleration.