Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

Annapurna Labs, an AWS company, is seeking a Senior Compiler Engineer III to join their Machine Learning team. This role focuses on developing and scaling a compiler for the world's largest ML workloads, specifically for AWS Machine Learning accelerators.

The position is part of the Neuron Compiler team, which is responsible for creating a deep learning compiler stack that optimizes models from frameworks like TensorFlow, PyTorch, and JAX to run efficiently on AWS accelerators. The team works on cutting-edge technology that powers AWS's ML infrastructure, including Inferentia and Trainium chips.

As a Sr. Machine Learning Compiler Engineer, you'll be involved in:

• Architecting and implementing business-critical compiler features
• Contributing to cutting-edge research in ML compilation
• Working with AWS ML services teams as a technical partner
• Participating in pre-silicon design and product development
• Building high-impact solutions for a large customer base

The role requires strong expertise in compiler development, with a preference for machine learning and AI accelerator experience. You'll be working in a startup-like environment within AWS, focusing on innovative solutions and performance optimization.

The position offers competitive compensation ranging from $151,300 to $261,500 based on location, plus equity and comprehensive benefits. You'll be part of a diverse, inclusive team that values work-life harmony and provides extensive opportunities for career growth and mentorship.

This is an excellent opportunity for experienced compiler engineers who want to work on cutting-edge ML technology while having a significant impact on AWS's machine learning infrastructure. The role requires physical presence in either Cupertino (preferred), Seattle, or Austin.