Amazon
Software Development Manager - NKI Compiler, AWS Neuron, Annapurna Labs
$213K–$288K
Cupertino, CA · Manager · Full-time · Posted Jul 21, 2026
- Experience
- 2+ years of engineering team management; 6+ years working directly within engineering teams; 4+ years designing or architecting systems.
About the job
Leads a team developing and delivering compiler optimizations and features for AWS Neuron’s NKI compiler, targeting Trainium hardware.
What you'll do
- Lead, hire, grow, and mentor compiler engineers while supporting team health and career development.
- Own NKI compiler execution, release delivery, technical direction, and roadmap.
- Collaborate across software, product, and hardware teams and guide resolution of complex technical issues.
What they ask for
- 2+ years of engineering team management experience.
- 6+ years of working directly within engineering teams experience.
- 4+ years of designing or architecting new and existing systems experience.
- Experience partnering with product or program management teams; understanding of compilers and strong software design and system-level coding skills.
Nice to have
- M.S. or Ph.D. in Computer Science or a related technical field.
Pay
212,700.00 - 287,700.00 USD annually (base salary)
Tools
AWS Neuron, NKI, Trainium, PyTorch, JAX, CPU, NPU, GPU
AI focus
Builds and optimizes a compiler that enables high-performance, productive ML workloads on Trainium.
Model-written summary · the original posting is authoritative