Machine Learning - Compiler Engineer, AWS Neuron, Annapurna Labs
Compensation
$165k – $223k /yr
Employment type
Full-time
Work setting
On-site
Location
Cupertino, CA
Schedule
Day shift
Posted
You'll be redirected to the employer's application page.
Job overview
Compensation is $165,000–$223,000 per year. The Machine Learning - Compiler Engineer, AWS Neuron, Annapurna Labs role is an onsite position based in Cupertino, CA. 00 - 223,600.00 USD annually, and this role supports the Annapurna Labs team within AWS, focusing on ML compiler development. This role exists to build the next-generation Neuron compiler for Trainium and Inferentia chips. It contributes to the team's goal of achieving optimal performance for diverse ML model families.
What you'll do
- Design and implement compiler optimization passes
- solve hard optimization problems
- collaborate with chip architects and ML teams
- work with open-source technologies like MLIR and StableHLO
- build innovative features.
What we're looking for
- Skills & competencies
- full-timeday shiftcupertino caannapurna labscompiler engineermlcompilerpytorchmliroptimization$165k-$223k
- Work arrangement
- Weekend coverage required
Benefits & perks
- Health insurance, 401(k) matching, paid time off, parental leave, sign-on payments, and restricted stock units (RSUs).
Why this role
Focus on ML compiler optimization for custom hardware using MLIR/StableHLO.
About the employer
Annapurna Labs (U.S.) Inc. is hiring for this role. Industry: Custom Computer Programming Services. Sector: 54.
Additional details
- Industry sector
- 54
- Industry
- Custom Computer Programming Services
- Occupation code
- 15-1252.00
You'll be redirected to the employer's application page.
Listing ID: 1d5f1178-13ca-4536-b3b8-48c97c0bde6a