Applied Scientist, ML Systems for AWS Neuron

Amazon; Cupertino (CA); On-site; USD 171,600 - 222,200; Full time

Benefits offered by this job

RSUs

Health insurance

401(k) matching

Paid time off

Job summary

Amazon’s AWS Neuron Science team is seeking an Applied Scientist to advance our software stack for Trainium and Inferentia accelerators. You will work with customers to identify adoption barriers and optimization opportunities, collaborating closely with engineering teams to implement innovative ML solutions.

The role engages with academia and research communities to push state-of-the-art machine-learning systems while contributing to a strategic growth area within AWS.

Qualifications

• PhD or Master’s degree in CS/EE/ML or related field.
• Experience with patents or publications at top-tier venues.
• Proficiency in Java, C++, Python or related languages.
• Strong background in algorithms, data structures and high‑performance computing.

Responsibilities

• Contribute to AI for systems and ML compiler initiatives on kernel and code generation and optimization.
• Develop and apply ML approaches for acceleration and compiler optimization.
• Support system robustness through validation and testing of ML workloads.
• Collaborate with external customers and engineering teams to deliver scalable ML solutions.

Skills

Java

C++

Python

Algorithms & data structures

Numerical optimization

Parallel or distributed computing

Education

PhD or Master’s degree

Tools

Patents/publications

Job description

Amazon’s AWS Neuron Science team is seeking an Applied Scientist to advance our software stack for Trainium and Inferentia accelerators. You will work with customers to identify adoption barriers and optimization opportunities, collaborating closely with engineering teams to implement innovative ML solutions.

The role engages with academia and research communities to push state-of-the-art machine-learning systems while contributing to a strategic growth area within AWS.