Software Development Engineer II, Annapurna Labs

Annapurna Labs, an Amazon subsidiary, is at the forefront of AWS innovation, developing custom Machine Learning accelerators for cloud computing. This role is within the Neuron Compiler Engineering team, which is crucial for enabling large-scale ML workloads in the cloud.

The position focuses on developing and optimizing the compiler infrastructure that transforms neural network models from frameworks like TensorFlow, PyTorch, and Jax into executable code for AWS custom hardware. As a Software Development Engineer II, you'll be responsible for creating efficient build, test, and release mechanisms for the Neuron compiler, while also working on distributed execution solutions to enhance customer workload performance.

This is an exciting opportunity to work at the intersection of machine learning and systems engineering, where you'll collaborate with compiler, hardware, and ML engineers in a startup-like environment. The role offers significant technical challenges across the entire stack, from low-level optimization to high-level system design.

The position comes with competitive compensation ranging from $129,300 to $223,600 based on location, plus equity and comprehensive benefits. You'll be working in either Boston or Seattle, contributing to AWS's machine learning infrastructure that powers some of the world's largest ML workloads.

The ideal candidate brings 3+ years of software development experience, strong system design skills, and a passion for solving complex technical challenges. You'll be expected to drive technical discussions, participate in code reviews, and work cross-functionally with various AWS teams to deliver high-impact solutions for our global customer base.

This role offers the unique opportunity to work on cutting-edge ML infrastructure while being part of Amazon's larger mission to push the boundaries of cloud computing. You'll have the chance to make significant contributions to products that enable AI/ML innovation across countless industries and applications.