ML Compiler Engineer II - Neuron Kernel Interface , Annapurna Labs

The AWS Neuron Compiler team is seeking talented compiler engineers to join their innovative efforts in developing cutting-edge deep learning compiler technology. This role focuses on optimizing application models across various domains, including Large Language Models and Vision, working with frameworks like PyTorch, TensorFlow, and JAX.

As an ML Compiler Engineer II, you'll be part of the Neuron Compiler Automated Reasoning Group, where you'll develop and maintain critical tooling to ensure the highest standards of compiler accuracy and reliability. You'll work with custom-built Machine Learning accelerators like Inferentia/Trainium, representing AWS's cutting-edge innovation in ML capabilities.

The position requires strong software development skills in C++/Python and a deep understanding of compiler development. You'll be working at the intersection of compiler technology and machine learning, helping to push the boundaries of what's possible in AI acceleration. The role offers the opportunity to work with state-of-the-art technology and contribute to solutions powering Generative AI applications.

The position offers competitive compensation ranging from $129,300 to $223,600 based on location, plus additional benefits including equity and comprehensive medical coverage. This is an excellent opportunity for someone passionate about compiler technology and machine learning to make a significant impact at one of the world's leading technology companies.

The role requires relocation to either Seattle (preferred), Cupertino, Austin, or Toronto, offering flexibility in location while maintaining the collaborative, in-person work environment essential for this technical role.