Machine Learning - Compiler Engineer II, Annapurna Labs

AWS Machine Learning accelerators are leading AWS's innovation in Generative AI tools. The role focuses on developing the AWS Neuron Software Development Kit (SDK), which includes an ML compiler and runtime that integrates with frameworks like PyTorch and TensorFlow. The Inferentia chip delivers industry-leading ML inference performance at the lowest cloud cost, while Trainium provides best-in-class ML training performance.

As a Machine Learning Compiler Engineer II on the AWS Neuron team, you'll be instrumental in developing a compiler handling the world's largest ML workloads. You'll work with cutting-edge technology, architecting features, publishing research, and collaborating with AWS ML services teams. The role involves pre-silicon design and bringing new products to market.

The team values work-life balance and offers flexible working hours. You'll join a diverse, inclusive environment with ten employee-led affinity groups and innovative benefits. The team emphasizes mentorship and knowledge sharing, with opportunities for career growth through challenging projects.

The position requires expertise in compiler optimization and software development, with a focus on machine learning infrastructure. You'll be part of Amazon Annapurna Labs, responsible for AWS silicon development, working alongside experts in silicon engineering, hardware design, and software development.

This role offers competitive compensation based on location and experience, with a comprehensive benefits package. You'll contribute to projects used by major customers like Snap, Autodesk, and Amazon's own services, making a significant impact on AWS's machine learning infrastructure.