Sr ML Compiler Engineer, Annapurna Labs

AWS Machine Learning accelerators are at the forefront of AWS innovation, focusing on building Generative AI solutions. The role is within the Amazon Annapurna Labs team, responsible for silicon development at AWS. As a Sr. Machine Learning Compiler Engineer III on the AWS Neuron team, you'll be developing a cutting-edge compiler stack for neural networks, working with frameworks like TensorFlow, PyTorch, and MXNET. The position involves optimizing performance of complex neural net models on custom-built AWS hardware.

The team is developing the AWS Neuron Software Development Kit (SDK), which includes an ML compiler, runtime, and integrates with popular ML frameworks. Their products, Inferentia and Trainium chips, deliver best-in-class ML inference and training performance at the lowest cost in cloud. These solutions are used by major customers like Snap, Autodesk, Amazon Alexa, and Amazon Rekognition.

You'll be working alongside some of the brightest minds in engineering, research, and product development, focused on creating a revolutionary toolchain for quantum leap performance improvements. The role requires strong technical communication skills and involves partnerships with AWS ML services teams. You'll contribute to pre-silicon design, product launches, and various innovative projects.

The position offers comprehensive benefits, competitive compensation based on location ($151,300-$261,500/year base pay), plus equity and other benefits. Amazon values work-life harmony and provides extensive career development opportunities, including mentorship and knowledge-sharing resources. The company embraces diversity through employee-led affinity groups and fosters an inclusive culture that celebrates differences.