Amazon
ML Kernel Performance Engineer, AWS Neuron, Annapurna Labs
$165K–$224K
Cupertino, CA · Full-time · Posted Sep 1, 2026
- Experience
- 3+ years of non-internship professional software development; 2+ years of non-internship systems design or architecture experience.
- Education
- Bachelor's degree in computer science or equivalent.
About the job
Designs and optimizes high-performance machine-learning kernels and compiler features for AWS Neuron accelerators, including customer model enablement.
What you'll do
- Design and implement compute kernels and compiler optimizations for Neuron architecture.
- Analyze kernel performance with profiling tools and resolve bottlenecks across Neuron hardware generations.
- Work with customers and cross-functional teams to enable and optimize machine-learning models on AWS accelerators.
What they ask for
- 3+ years of non-internship professional software development experience.
- 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience.
- Experience programming with at least one software programming language.
Nice to have
- 3+ years of full software development life cycle experience.
- Bachelor's degree in computer science or equivalent.
Pay
165,200.00 - 223,600.00 USD annually (base salary)
Tools
AWS Neuron, Inferentia, Trainium, PyTorch
AI focus
Accelerates deep learning and GenAI workloads through kernel optimization, compiler features, and model enablement on AWS machine-learning accelerators.
Model-written summary · the original posting is authoritative