Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs

Product

AWS Machine Learning accelerators are at the forefront of AWS innovation and enable the building of Generative AI on AWS. The Inferentia chip delivers best‑in‑class ML inference performance at the lowest cost. Trainium offers unparalleled ML training performance with the most teraflops (TFLOPS) of compute power in the cloud. The AWS Neuron Software Development Kit (SDK) integrates a compiler, runtime, and native support into popular ML frameworks such as PyTorch, TensorFlow, and MXNet. AWS Neuron and Inferentia are deployed at scale with customers including Snap, Autodesk, Amazon Alexa, Amazon Rekognition, and many others.

The Team

The Amazon Annapurna Labs team is responsible for silicon development at AWS, covering disciplines such as silicon engineering, hardware design, verification, software, and operations. The AWS Neuron team focuses on optimizing complex neural‑net models on custom AWS hardware by building a deep‑learning compiler stack that transforms framework‑based models into efficient executable code.

About the Role

As a Sr. Machine Learning Compiler Engineer III on the AWS Neuron team, you will play a thought‑leader role in developing and scaling a compiler to handle the world’s largest ML workloads. You will architect and implement business‑critical features, publish cutting‑edge research, mentor a talented engineering team, collaborate with AWS ML services teams, and participate in pre‑silicon design and product launches. A background in ML and AI accelerators is preferred but not required.

Location Requirements

Candidates must be located or willing to relocate to Cupertino (preferred), Seattle, Austin, or Toronto.

Qualifications

• 5+ years of professional, non‑internship software development experience
• 5+ years of programming in at least one software language
• 5+ years of design or architecture experience, including design patterns, reliability, and scaling of new and existing systems
• 5+ years of full software development life cycle experience, covering coding standards, code reviews, source control, build processes, testing, and operations
• Experience as a mentor, tech lead, or engineering team leader

• Bachelor's degree in computer science or equivalent

Team Culture

Inclusive Team Culture: AWS embraces diversity and promotes inclusion through 10 employee‑led affinity groups and global learning experiences like CORE and AmazeCon.

Work/Life Balance: The team values flexible schedules and encouraging a personal‑professional balance that supports long‑term well‑being.

Mentorship & Career Growth: New members receive strong mentorship and project assignments that foster growth into well‑rounded professionals.

EEO Statement

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Compensation

Base pay ranges from $151,300 to $261,500 per year, depending on geographic market. Additional equity, sign‑on payments, and benefits may apply. For more information on total compensation, visit

#J-18808-Ljbffr