Join the innovative Annapurna Labs team at AWS 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 a groundbreaking SDK that optimizes neural net model performance on AWS Inferentia and Trainium custom chips. You'll be part of a team democratizing access to cutting-edge ML infrastructure, working with technologies like TensorFlow, PyTorch, and MXNet.
As a senior engineer, you'll architect and implement critical features, publish research, and mentor experienced engineers. The role combines technical leadership with hands-on development, requiring expertise in compiler development and deep learning systems. You'll work in an agile, startup-like environment where innovation is constant and impact is immediate.
AWS offers an inclusive culture with 10 employee-led affinity groups and various benefits supporting work-life balance. The team values knowledge sharing, mentorship, and career growth, assigning projects that help members develop into well-rounded professionals. AWS embraces diverse experiences and backgrounds, encouraging applications from candidates with non-traditional paths.
Key responsibilities include solving complex technical challenges, designing scalable solutions, and collaborating across teams to drive business decisions. You'll work on high-impact projects that directly affect AWS's global customer base, while maintaining high standards for software quality and reliability.
The position offers the opportunity to work with cutting-edge ML infrastructure, contribute to innovative products, and shape the future of cloud computing. You'll be part of AWS's culture of inclusion, guided by their 16 Leadership Principles, which emphasize seeking diverse perspectives, continuous learning, and earning trust.