Sr. ML Compiler Engineer - Automated Reasoning Science, Annapurna Labs

The AWS Neuron Compiler team at Amazon is seeking an experienced Senior ML Compiler Engineer to join their cutting-edge development efforts. This role focuses on building a state-of-the-art deep learning compiler stack that optimizes application models across various domains, including Large Language Models and Vision, working with frameworks like PyTorch, TensorFlow, and JAX.

The position is part of the Neuron Compiler Automated Reasoning Group, where you'll be instrumental in developing and maintaining advanced tooling to enhance the compiler's accuracy and reliability. You'll work with custom-built ML accelerators like Inferentia/Trainium, representing AWS's innovative approach to advanced machine learning capabilities.

The role requires strong expertise in C++ and Python programming, with a deep understanding of compiler development. You'll be working at the intersection of compiler technology and machine learning, contributing to AWS's position at the forefront of AI infrastructure. The team emphasizes a scientific approach to software engineering, pushing the boundaries of what's possible in compiler technology.

This is an excellent opportunity for someone passionate about both compiler technology and machine learning, offering competitive compensation ranging from $151,300 to $261,500 based on location, plus additional benefits including equity, sign-on bonuses, and comprehensive medical coverage. The position requires physical presence in either Seattle (preferred), Cupertino, Austin, or Toronto, offering the chance to work with some of the best minds in the field while solving complex technical challenges that impact AWS's global customer base.