Machine Learning - Compiler Engineer II, Annapurna Labs

AWS Machine Learning accelerators are leading AWS innovation in Generative AI development. The team is developing Inferentia chip for ML inference and Trainium for ML training, supported by the AWS Neuron SDK. As a Machine Learning Compiler Engineer II on the AWS Neuron team, you'll be crucial in developing a compiler for large-scale ML workloads. The role involves working with cutting-edge technology at Annapurna Labs, Amazon's silicon development arm.

The position requires expertise in compiler optimization and machine learning systems. You'll work on converting neural network descriptions from frameworks like TensorFlow and PyTorch into executable code. The team comprises top engineering talent focused on creating revolutionary performance improvements in ML computing.

Key responsibilities include developing compiler architecture, implementing critical features, publishing research, and collaborating with AWS ML services teams. You'll be involved in pre-silicon design and new product launches. The role offers significant growth opportunities in a rapidly evolving field.

The team values work-life balance and fosters an inclusive culture through employee-led affinity groups and diversity initiatives. Amazon's leadership principles emphasize seeking diverse perspectives and continuous learning. The position offers competitive compensation, comprehensive benefits, and a supportive environment for professional development.

This role is perfect for someone passionate about compiler technology and machine learning, offering the chance to work on transformative technology that powers major customers like Snap, Autodesk, and Amazon's own services.