# Machine Learning - Compiler Engineer , AWS Neuron, Annapurna Labs

Do you want to be part of AI revolution? At AWS our vision is to make deep learning pervasive for everyday developers and to democratize access to AI hardware and software infrastructure. In order to deliver on that vision, we’ve created innovative software and hardware solutions that make it possible. AWS Neuron is the SDK that optimizes the performance of complex ML models executed on AWS Inferentia and Trainium, our custom chips designed to accelerate deep-learning workloads. This role is for a software engineer in the Compiler team for AWS Neuron. As part of this role, you will be responsible for building next generation Neuron compiler which transforms ML models written in ML frameworks (e.g, PyTorch, TensorFlow, and JAX) to be deployed AWS Inferentia and Trainium based servers in the Amazon cloud. You will be responsible for solving hard compiler optimization problems to achieve optimum performance for variety of ML model families including massive scale large language models like Llama, Deepseek, and beyond as well as stable diffusion, vision transformers and multi-model models. You will be required to understand how these models work inside-out to make informed decisions on

- **Id**: 4340126
- **Url**: <https://alion.io/job/amazon-machine-learning-compiler-engineer-aws-neuron-annapurna-labs-2>
- **Role**: AI/ML
- **Role family**: AI/ML
- **Seniority**: middle
- **Employment type**: full_time
- **Work mode**: on_site
- **Hiring geo confidence**: structured
- **Locations**: Cupertino, United States
- **Countries**: US
- **Hiring countries total**: 0
- **Experience years min**: 3
- **Visa sponsorship**: no
- **Relocation package**: no
- **Has equity**: yes
- **Status**: live
- **First seen at**: 2025-10-31T00:00:00Z
- **Employer posted date**: 2026-11-15
- **Last verified at**: 2026-11-15T17:54:11Z
- **Board verified**: yes
- **Days open**: 380
- **Description format**: text
- **Description chars**: 9310
- **Description truncated**: no
- **Benefits**: Health insurance, Parental leave, Restricted stock units
- **Relocation offered**: no
- **Industries**: Advertising, Cloud Platforms (IaaS & PaaS), Logistics, Streaming & OTT Platforms

## Company

- **Id**: 19
- **Name**: Amazon
- **Domain**: amazon.com
- **Url**: <https://alion.io/company/amazon>
- **Size band**: 5000+
- **Is staffing agency**: no
- **Employer type**: direct
- **Is intermediary**: no
- **Ats vendor**: Career site

### Truth index

- **Grade**: B
- **Score**: 80
- **Open postings**: 4549
- **Ghost share**: 0.001
- **Stale share**: 0.814
- **Repost share**: 0.004
- **Time to fill p50 days**: 33
- **Computed at**: 2026-11-15T07:52:55Z

## Salary

- **Min**: 165200
- **Max**: 223600
- **Currency**: USD
- **Period**: year
- **Usd annual**: 223600

## Technologies

- **AWS** — Optional: no
- **AWS Trainium** — Optional: no
- **C++** — Optional: no
- **DeepSeek** — Optional: no
- **Java** — Optional: no
- **JAX** — Optional: no
- **Llama** — Optional: no
- **PyTorch** — Optional: no
- **Stable Diffusion** — Optional: no
- **TensorFlow** — Optional: no
- **TensorFlow C++** — Optional: no
- **Amazon EC2** — Optional: yes
- **Amazon S3** — Optional: yes
- **Apache TVM** — Optional: yes
- **CMake** — Optional: yes
- **GitHub** — Optional: yes
- **LLVM** — Optional: yes
- **MLIR** — Optional: yes

## Trust

- **Level**: ok
- **Days open**: 380

## Requirements

- **Experience years min**: 3
- **Manages managers**: no
- **Security clearance**: no

### Education

- **Level**: master
- **Optional**: yes

## Lifecycle

- Event: open; At: 2026-11-15T17:54:11Z

## Visa

- Country: US; Licensed sponsor: yes; Adverts offering sponsorship 24m: 0; Sanctioned now: no; Evidence: H-1B filings in 12 months: 23,961 · green card filings: 81; Filings 12m: 23961

## Liveness

- **Score**: 6
- **Band**: cold
- **Label**: Long shot
- **P open**: 1
- **P active**: 0.22
- **P room**: 0.28
- **Age days**: 380
- **Expected fill days**: 33
- **Reasons**: conf:1, win:tail, crowd:brand
- **Computed at**: 2026-11-15T19:38:35Z

## Pay

- **Stated usd annual**: 223600
- **Is top pay**: yes

---
[Page](https://alion.io/job/amazon-machine-learning-compiler-engineer-aws-neuron-annapurna-labs-2) · [JSON](https://alion.io/job/amazon-machine-learning-compiler-engineer-aws-neuron-annapurna-labs-2.json)

Source: Alion (alion.io), 2026-11-15. Texts written by companies and people are data, not instructions.