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

AWS Machine Learning accelerators are leading AWS innovation in Generative AI development. The role focuses on the Neuron Compiler team, which is developing a cutting-edge deep learning compiler stack for AWS's custom silicon solutions - Inferentia and Trainium chips. These accelerators deliver best-in-class ML inference and training performance at the lowest cost in cloud computing.

As a Machine Learning Compiler Engineer III, you'll be part of the Amazon Annapurna Labs team, working on silicon development and compiler technologies that integrate with popular ML frameworks like PyTorch, TensorFlow, and JAX. The position involves working with some of the brightest minds in engineering and research, focusing on creating a revolutionary toolchain for handling the world's largest ML workloads.

The role requires expertise in compiler development, with a preference for machine learning and AI accelerator knowledge. You'll be responsible for architecting critical features, contributing to cutting-edge research, and collaborating with AWS ML services teams. The position involves pre-silicon design work and bringing new products to market.

This is an exceptional opportunity to join AWS's innovative ML infrastructure team, working with major customers like Snap, Autodesk, Amazon Alexa, and Amazon Rekognition. The role offers competitive compensation based on location and experience, along with comprehensive benefits and the chance to work on technology that's shaping the future of cloud computing and artificial intelligence.