Back to search

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

Annapurna Labs (U.S.) Inc. · Cupertino, CA · Annapurna Labs, AWS Neuron

Compensation

$193k – $261k /yr

Employment type

Full-time

Work setting

On-site

Location

Cupertino, CA

Schedule

Day shift

Posted

Apply for this job

You'll be redirected to the employer's application page.

Job overview

Compensation is $193,000–$261,000 per year. The Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs role is an onsite position based in Cupertino, CA. 00 - 261,500.00 USD annually, and this role supports the Annapurna Labs team within AWS, focusing on ML compiler development. This role exists to lead the development and scaling of a compiler for large ML workloads. It contributes to the team's goal of creating a high-performance toolchain for Trainium and Inferentia chips.

What you'll do

  • Architect and implement business-critical compiler features
  • mentor engineers
  • partner with AWS ML services teams
  • support pre-silicon design
  • publish research
  • drive compiler performance and usability.

What we're looking for

Skills & competencies
full-time
day shift
cupertino ca
annapurna labs
compiler engineer
ml
compiler
pytorch
tensorflow
performance
$193k-$261k
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 architecture and scaling for large-scale ML workloads.

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
Apply for this job

You'll be redirected to the employer's application page.

Browse more jobs

Listing ID: 5d25460e-e347-4aba-b264-0ed916aadfba

    Sr. Machine Learning - Compiler Engineer III,… | CollabWORK