ML Compiler Engineer II - Automated Reasoning Science, Annapurna Labs

The AWS Neuron Compiler team is seeking a skilled ML Compiler Engineer II to join their Automated Reasoning Group. This role focuses on developing a cutting-edge deep learning compiler stack that optimizes application models across various domains, including Large Language and Vision models from frameworks like PyTorch, TensorFlow, and JAX. The position involves working with AWS's custom-built Machine Learning accelerators (Inferentia/Trainium) that power advanced ML capabilities and Generative AI solutions.

The role requires strong software development skills in C++/Python and a background in compiler development. You'll be responsible for developing and maintaining tooling to enhance the Neuron Compiler's accuracy and reliability, leading efforts in building fuzzers and specification synthesis tooling for their LLVM-based compiler. The position offers a competitive salary range of $129,300 to $223,600 based on location and experience.

Working at Amazon, you'll be part of a team with a strong science focus, pushing the boundaries of what's possible in compiler technology. The position offers the opportunity to work with state-of-the-art ML accelerators and contribute to the development of next-generation AI infrastructure. The role requires a minimum of 3 years of experience in leading design or architecture and 2 years in compiler development.

This is an excellent opportunity for someone passionate about compiler technology and machine learning, offering the chance to work on cutting-edge technology that powers AWS's AI capabilities. The position includes comprehensive benefits and the opportunity to work with some of the most advanced ML infrastructure in the industry. The role requires in-person work at one of several locations: Seattle (preferred), Cupertino, Austin, or Toronto.