ML Compiler Engineer I, Annapurna Labs

Annapurna Labs, now fully integrated into AWS after its 2015 acquisition, is seeking a Machine Learning Compiler Engineer I to join their innovative team. This role focuses on developing and scaling compilers for large-scale ML workloads on custom 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 for AWS accelerators. The team's work directly impacts AWS's ML infrastructure, supporting major customers like Snap, Autodesk, and Amazon Alexa.

As an ML Compiler Engineer I, you'll be working on groundbreaking compiler development projects, implementing critical features, and contributing to cutting-edge research. The role requires strong technical communication skills as you'll partner with AWS ML services teams and be involved in pre-silicon design and product development.

The position offers competitive compensation ranging from $99,500 to $200,000 based on location, plus additional benefits including equity, sign-on payments, and comprehensive medical coverage. This is an excellent opportunity for recent graduates with compiler development experience to join one of tech's most innovative teams.

The role requires relocation to either Cupertino (preferred), Seattle, or Toronto, offering the chance to work in major tech hubs. You'll be part of a team developing next-generation ML acceleration technologies, working alongside experienced engineers and researchers in a collaborative environment.

This position is ideal for someone passionate about compiler technology and machine learning, offering the opportunity to work on large-scale projects that directly impact AWS's ML infrastructure. The role combines technical challenges with business impact, making it an excellent starting point for a career in advanced computing systems.