Software Development Engineer III, Annapurna Labs

Annapurna Labs, an Amazon company, is seeking a Senior Software Development Engineer to join their Neuron Compiler Engineering team. This role is at the forefront of AWS innovation, working on custom Machine Learning accelerators that power AWS's Generative AI capabilities. The position focuses on developing a sophisticated deep learning compiler stack that transforms neural network models from frameworks like TensorFlow, PyTorch, and Jax into executable code.

As a Senior SDE, you'll be instrumental in designing and implementing solutions for distributed execution of the Neuron compiler, enhancing efficiency and reliability of build systems, and optimizing test and release mechanisms. This role offers a unique opportunity to work with cutting-edge ML hardware and software, directly impacting how the world's largest ML workloads run in the cloud.

The position combines technical depth with leadership, requiring collaboration across diverse teams including compiler, hardware, and ML engineers. You'll work in a dynamic, startup-like environment where your contributions directly affect Amazon's global customer base. The role demands expertise in software architecture, system design, and technical communication, with opportunities to mentor others and lead technical initiatives.

This is an excellent opportunity for experienced engineers passionate about machine learning infrastructure, compiler technology, and distributed systems. The role offers competitive compensation ranging from $151,300 to $261,500 based on location, plus equity and comprehensive benefits. Working at Amazon's Annapurna Labs means being at the intersection of hardware and software innovation, with the resources of a tech giant and the agility of a startup.