Amazon
Applied Scientist, AWS Neuron Science team
$172K–$222K
Cupertino, CA · Full-time · Posted May 27, 2026
- Experience
- PhD, or Master's degree and 4+ years of CS, CE, ML, or related field experience.
- Education
- PhD, or Master's degree
About the job
Develop and improve the AWS Neuron software stack for Trainium and Inferentia, addressing customer adoption barriers and advancing ML systems.
What you'll do
- Identify customer adoption barriers and optimization opportunities, and collaborate with engineering teams to implement solutions.
- Develop ML/compiler techniques, robust validation tools, and efficient kernels for ML accelerator architectures.
- Engage with academic and research communities to advance ML systems.
What they ask for
- PhD, or Master's degree and 4+ years of CS, CE, ML, or related field experience.
- Experience in patents or publications at top-tier peer-reviewed conferences or journals.
- Programming experience in Java, C++, Python, or a related language.
- Experience in algorithms and data structures, parsing, numerical optimization, data mining, parallel and distributed computing, or high-performance computing.
Nice to have
- Experience building machine learning models or developing algorithms for business application.
- Experience in professional software development.
Pay
171,600.00 - 222,200.00 USD annually (base salary)
Tools
Trainium, Inferentia, Java, C++, Python, ML compiler
AI focus
Develop and apply ML/RL approaches for kernel and code generation and optimization; create compiler techniques for ML workloads; build tools for accuracy and reliability validation; develop efficient kernels for ML accelerators.
Model-written summary · the original posting is authoritative