Amazon
Senior ML Compiler Engineer, Neuron
$168K–$262K
Seattle, WA +1 · Senior · Full-time · Posted Sep 30, 2026
- Location
- Seattle, WA
Cupertino, CA - Experience
- 5+ years for each of the three stated experience requirements; mentoring, tech lead, or engineering-team leadership experience is also required.
- Education
- Master's degree or Ph.D. in computer science, computer engineering, or related field is preferred.
About the job
Build and scale AWS Neuron compiler features for Trainium, partner with ML teams to optimize workloads, and contribute to hardware design and engineering leadership.
What you'll do
- Design and build compiler features across frontends, intermediate representations, optimization passes, and code generation.
- Own customer-critical features from scoping and design through production delivery and customer support; work with ML teams to translate workloads into compiler needs.
- Provide pre-silicon feedback, review code and designs, publish research, and mentor engineers.
What they ask for
- 5+ years of non-internship professional software development experience.
- 5+ years programming with at least one software programming language.
- 5+ years leading design or architecture of new and existing systems, including design patterns, reliability, and scaling.
- Experience as a mentor, tech lead, or engineering team leader.
Nice to have
- 5+ years of full software development life cycle experience.
- Master's degree or Ph.D. in computer science, computer engineering, or a related field.
Pay
193,300.00 - 261,500.00 USD annually (base salary); 168,100.00 - 227,400.00 USD annually (base salary)
Tools
Neuron, Trainium, PyTorch, JAX, Neuron Kernel Interface (NKI)
AI focus
Develop compiler software for ML workloads and AWS Trainium accelerators; help ML teams optimize and run workloads on hardware.
Model-written summary · the original posting is authoritative