Amazon
Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs
$193K–$262K
Cupertino, CA · Senior · Full-time · Posted Oct 6, 2026
- Experience
- 5+ years for each of the three stated experience requirements; experience as a mentor, tech lead, or engineering team lead also required. No alternatives stated.
- Education
- Bachelor's degree in computer science or equivalent (preferred).
About the job
Build and improve the AWS Neuron compiler that optimizes machine-learning models for deployment on AWS Inferentia and Trainium accelerators.
What you'll do
- Design, implement, test, deploy, and maintain compiler software, including optimization and verification passes and tools for numerical-error analysis.
- Collaborate with chip architects, runtime and OS engineers, scientists, and ML teams to optimize model deployment and performance.
- Diagnose compiler defects, contribute to product development, and engage with open-source communities and other stakeholders.
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 lead.
Nice to have
- 5+ years of full software development life cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
- Bachelor's degree in computer science or equivalent.
Pay
193,300.00 - 261,500.00 USD annually (base salary)
Tools
AWS Neuron, AWS Inferentia, Trainium, C++, Java, PyTorch, TensorFlow, JAX
AI focus
Optimize compilation and deployment of machine-learning workloads, including large language models, stable diffusion, vision transformers, and multimodal models, on AWS accelerators.
Model-written summary · the original posting is authoritative