Software Development Engineer III, Annapurna Labs

Annapurna Labs, a crucial part of AWS's innovation ecosystem, is seeking a Senior Software Development Engineer to join their Neuron Compiler Engineering team. This role sits at the intersection of Machine Learning and cloud computing, focusing on developing infrastructure for a compiler that enables the world's largest ML workloads to run efficiently in the cloud.

The position offers an exciting opportunity to work with AWS's custom-built hardware and develop a deep learning compiler stack that handles neural network descriptions from frameworks like TensorFlow, PyTorch, and Jax. As a Senior SDE, you'll be responsible for creating distributed execution solutions for the Neuron compiler and implementing efficient build, test, and release mechanisms.

The role combines technical depth with cross-functional collaboration, requiring interaction with compiler, hardware, and ML engineers. You'll work in a dynamic, startup-like environment where your contributions directly impact AWS's global customer base. The position offers competitive compensation ranging from $151,300 to $261,500 per year, depending on location, plus additional benefits including equity, sign-on payments, and comprehensive medical coverage.

This is an excellent opportunity for experienced engineers who want to work at the forefront of ML acceleration and cloud computing, solving complex technical challenges that haven't been addressed before. The role requires strong technical communication skills and the ability to partner with AWS ML services teams to bring new products and features to market.

Working at Amazon's Annapurna Labs means joining a team responsible for silicon development at AWS, covering multiple disciplines including silicon engineering, hardware design and verification, software, and operations. The position offers the chance to make a significant impact on how machine learning workloads are optimized and executed in the cloud, while working with cutting-edge technology and some of the most challenging problems in the field.