Software Development Engineer II, Annapurna Labs

Annapurna Labs, an AWS subsidiary, is at the forefront of Machine Learning acceleration technology, developing custom hardware accelerators for AWS's Generative AI initiatives. The Neuron Compiler Engineering team is seeking a Software Development Engineer to join their innovative efforts in optimizing ML workloads for cloud deployment. This role focuses on developing the infrastructure for a compiler that enables efficient execution of large-scale machine learning models on AWS hardware.

The position involves working with the AWS Neuron team, which specializes in developing a deep learning compiler stack that transforms neural network models from frameworks like TensorFlow, PyTorch, and Jax into optimized executable code. As an SDE II, you'll be responsible for creating and implementing solutions for build systems, testing frameworks, and release processes that enhance the compiler's efficiency and reliability.

The role offers an exciting opportunity to work in a startup-like environment within AWS, where you'll collaborate with cross-functional teams including compiler engineers, hardware specialists, and ML experts. You'll be tackling unprecedented technical challenges across the entire stack, from low-level optimizations to high-level system architecture. The position requires strong technical communication skills and the ability to work effectively with various AWS ML services teams to bring new products and features to market.

This is an excellent opportunity for engineers passionate about machine learning infrastructure, compiler technology, and distributed systems. The role offers competitive compensation, including base salary, equity, and comprehensive benefits. Working at Amazon's Seattle location, you'll be part of a team that's directly impacting how the world's largest ML workloads are executed in the cloud.