Sr. Deep Learning Compiler Engineer III, AWS Neuron, Annapurna Labs

Join AWS Neuron team at Amazon as a Senior Deep Learning Compiler Engineer, where you'll work on cutting-edge products at the intersection of machine learning, high-performance computing, and distributed architectures. AWS Neuron is the SDK that optimizes neural net model performance on AWS Inferentia and Trainium custom chips. You'll be part of Annapurna Labs, whose vision is to democratize deep learning for everyday developers.

As a senior engineer, you'll architect and implement business-critical features, publish cutting-edge research, and mentor experienced engineers. The role involves working with AWS ML services teams, participating in pre-silicon design, and bringing new products/features to market. You'll be developing and scaling the compiler to handle the world's largest ML workloads.

The position offers a unique learning culture where teams remain small and agile despite working on large-scale projects. You'll collaborate with diverse teams to impact our global customer base directly. The role combines technical leadership with hands-on development, requiring both deep technical expertise and strong communication skills.

AWS values work-life balance and fosters an inclusive team culture with ten employee-led affinity groups across 190 global chapters. The company embraces diverse experiences and perspectives, following their 16 Leadership Principles that emphasize seeking diverse viewpoints and earning trust.

You'll be working with the Neuron SDK, which includes a compiler, run-time, and debugger, integrated with major frameworks like Tensorflow, PyTorch, and MXNet. This is an opportunity to work on innovative solutions that make deep learning accessible and efficient for AWS customers worldwide.