Applied Scientist

The AWS Neuron Science Team is looking for talented scientists to enhance our software stack, accelerating customer adoption of Trainium and Inferentia accelerators. In this role, you will work directly with external and internal customers to identify key adoption barriers and optimization opportunities. You’ll collaborate closely with our engineering teams to implement innovative solutions and engage with academic and research communities to advance state‑of‑the‑art ML systems.

We actively work on these areas:

• AI for Systems:
  Developing and applying ML/RL approaches for kernel/code generation and optimization
• Machine Learning Compiler:
  Creating advanced compiler techniques for ML workloads
• System Robustness:
  Building tools for accuracy and reliability validation
• Efficient Kernel Development:
  Designing high‑performance kernels optimized for our ML accelerator architectures

About the team: AWS Neuron is the software that runs on Trainium and Inferentia, integrating ML compiler and native support into popular frameworks. Our products are deployed at scale with external customers such as Anthropic and Databricks, as well as internal customers like Alexa, Amazon Bedrock, Amazon Robotics, Amazon Ads, and Amazon Rekognition.

• PhD in math, statistics, engineering, or an equivalent quantitative discipline
• Experience building machine learning models or developing algorithms for business applications
• Having publications at top‑tier peer‑reviewed conferences or journals
• Experience programming in Java, C++, Python, or a related language

• Experience building machine learning models for business applications
• Experience in professional software development

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Los Angeles County applicants:
Job duties for this position include working safely and cooperatively with other employees, supervisors, and staff; adhering to standards of excellence despite stressful conditions; communicating effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service; and following all federal, state, and local laws and company policies. Criminal history may have a direct and adverse relationship with some of the material job duties of this position.

Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider qualified applicants with arrest and conviction records.

Base salary range:  –  USD annually.

Amazon also offers comprehensive benefits, including health insurance (medical, dental, vision, prescription, basic life & AD&D insurance, and optional supplemental life plans), employee assistance program, mental health support, medical advice line, flexible spending accounts, adoption and surrogacy reimbursement coverage, 401(k) matching, paid time off, and parental leave.