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 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 Automated Reasoning Group, where you'll be developing advanced tooling for the Neuron Compiler's accuracy and reliability.

The role requires expertise in C++/Python programming and compiler development, with a focus on LLVM-based systems. You'll be working with AWS's custom-built Machine Learning accelerators (Inferentia/Trainium) that power innovative solutions like Generative AI. The position offers a competitive salary range of $129,300 to $223,600, depending on location and experience, plus comprehensive benefits including equity and sign-on bonuses.

As part of Amazon's AWS division, you'll be working at the forefront of machine learning infrastructure, helping to build and optimize the tools that power next-generation AI applications. The role requires a strong technical background, with preferred qualifications including a Masters or PhD in computer science and experience with deep learning frameworks. You'll be joining a science-focused team that pushes the boundaries of what's possible in compiler technology and machine learning optimization.

The position offers the opportunity to work in major tech hubs including Seattle, Cupertino, Austin, or Toronto, with a collaborative team environment and access to cutting-edge technology. This is an excellent opportunity for someone passionate about compiler development, machine learning, and high-performance computing to make a significant impact on AWS's machine learning infrastructure.