Machine Learning - Compiler Engineer II, Annapurna Labs

Amazon
Cupertino, CA, USA / Boston, MA, USA / Seattle, WA, USA2026-05-13ONSITE

About the job

Machine Learning Compiler Engineer II on the AWS Neuron team, you will be supporting the ground-up development and scaling of a compiler to handle the world's largest ML workloads. Architecting and implementing business-critical features, publish cutting-edge research, and contributing to a brilliant team of experienced engineers excites and challenges you. You will leverage your technical communications skill as a hands-on partner to AWS ML services teams and you will be involved in pre-silicon design, bringing new products/features to market, and many other exciting projects.

Responsibilities

No responsibilities listed.

Qualifications

Minimum

3+ years of non-internship professional software development experience

2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience

Experience programming with at least one software programming language

3+ years of non-internship professional software development, or 3+ years of software development experience

2+ years of experience architecting and optimizing compilers

Proficiency with 1 or more of the following programming languages: C++ (preferred), C, Python

Preferred

3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience

Bachelor's degree in computer science or equivalent

PhD in computer science, computer engineering, or related field, or MS degree

Experience with multiple toolchains and Instruction Set Architectures

Proficiency with resource management, scheduling, code generation, and compute graph optimization

Experience optimizing Tensorflow, PyTorch or MxNET deep learning models