Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs

AWS Machine Learning accelerators are at the forefront of AWS innovation, powering Generative AI solutions on AWS. The role is within the Neuron Compiler team at Amazon's Annapurna Labs, focusing on developing a cutting-edge deep learning compiler stack for AWS's ML accelerators - Inferentia and Trainium chips. These accelerators deliver best-in-class ML inference and training performance in the cloud.

As a Senior Machine Learning Compiler Engineer, you'll be working on transforming neural network descriptions from frameworks like TensorFlow, PyTorch, and JAX into optimized executable code. The position involves collaborating with some of the brightest minds in engineering and research to create a revolutionary toolchain for maximum performance.

The role requires expertise in compiler development, with a preference for Machine Learning and AI accelerator experience. You'll be responsible for architecting critical features, contributing to research, and working closely with AWS ML services teams. The position also involves pre-silicon design work and bringing new products to market.

The team's work directly impacts major customers like Snap, Autodesk, Amazon Alexa, and Amazon Rekognition. This is an opportunity to shape the future of machine learning infrastructure at scale, working with cutting-edge technology in a team of exceptional engineers.

The position offers competitive compensation ranging from $151,300 to $261,500 based on location, plus equity, sign-on bonuses, and comprehensive benefits. The role requires location in or willingness to relocate to Cupertino (preferred), Seattle, or Austin.