Sr ML Compiler Engineer, Annapurna Labs

AWS Machine Learning accelerators are at the forefront of AWS innovation, developing cutting-edge hardware and software solutions for ML inference and training. The role focuses on developing a compiler stack for AWS Neuron that optimizes neural network performance on custom AWS hardware.

As a Sr. Machine Learning Compiler Engineer III on the AWS Neuron team, you'll be responsible for:

• Leading the ground-up development and scaling of a compiler for large-scale ML workloads
• Architecting and implementing business-critical features
• Publishing cutting-edge research
• Mentoring experienced engineers
• Partnering with AWS ML services teams
• Contributing to pre-silicon design and new product features

The position is part of Amazon Annapurna Labs, responsible for silicon development at AWS, covering silicon engineering, hardware design and verification, software and operations. The team works with frameworks like TensorFlow, PyTorch, and MXNET to optimize neural network performance.

The role offers:

• Opportunity to work with cutting-edge ML hardware and software
• Collaboration with industry experts
• Work-life harmony
• Career growth and mentorship opportunities
• Inclusive team culture
• Comprehensive benefits package

The position requires strong technical leadership, software development expertise, and system architecture experience. Background in Machine Learning and AI accelerators is preferred but not required.

This is an excellent opportunity for an experienced engineer looking to make significant contributions to the future of ML hardware and software optimization at AWS.