Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs
Description For Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs
AWS Machine Learning accelerators are leading AWS innovation in Generative AI development. The role focuses on the Neuron Compiler team, which is developing a cutting-edge deep learning compiler stack for AWS's custom silicon solutions - Inferentia and Trainium chips. These accelerators deliver best-in-class ML inference and training performance at the lowest cost in cloud computing.
As a Machine Learning Compiler Engineer III, you'll be part of the Amazon Annapurna Labs team, working on silicon development and compiler technologies that integrate with popular ML frameworks like PyTorch, TensorFlow, and JAX. The position involves working with some of the brightest minds in engineering and research, focusing on creating a revolutionary toolchain for handling the world's largest ML workloads.
The role requires expertise in compiler development, with a preference for machine learning and AI accelerator knowledge. You'll be responsible for architecting critical features, contributing to cutting-edge research, and collaborating with AWS ML services teams. The position involves pre-silicon design work and bringing new products to market.
This is an exceptional opportunity to join AWS's innovative ML infrastructure team, working with major customers like Snap, Autodesk, Amazon Alexa, and Amazon Rekognition. The role offers competitive compensation based on location and experience, along with comprehensive benefits and the chance to work on technology that's shaping the future of cloud computing and artificial intelligence.
Responsibilities For Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs
- Support ground-up development and scaling of a compiler for large ML workloads
- Architect and implement business-critical features
- Publish cutting-edge research
- Partner with AWS ML services teams
- Participate in pre-silicon design
- Bring new products/features to market
Requirements For Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs
- 5+ years of leading design or architecture experience
- 2+ years of experience in developing compiler features and optimizations
- Proficiency with C++ (preferred), C, Python
- Must be willing to relocate to Cupertino, Seattle, or Austin
Benefits For Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs
- Medical Insurance
- Financial Benefits
- Employee Benefits
Jobs Related To Amazon Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs
