Applied Scientist, AWS Neuron Science team

Description

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, 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.

As part of a strategic growth area for AWS, you’ll work alongside distinguished engineers and scientists in an exciting and impactful environment.

• 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

AWS Neuron is the software stack of Amazon’s Trainium and Inferentia machine‑learning chips. Inferentia delivers best‑in‑class inference performance at the lowest cost, while Trainium delivers best‑in‑class training performance. Neuron provides a compiler and native integration into popular ML frameworks and is used at scale by external customers such as Anthropic and Databricks, as well as internal applications like Alexa, Amazon Bedrock, Amazon Robotics, Amazon Ads, and Amazon Rekognition.

• PhD, or Master’s degree and 4+ years of experience in computer science, electrical engineering, machine learning, or a related field.
• Experience with patents or publications at top‑tier peer‑reviewed conferences or journals.
• Programming experience in Java, C++, Python, or a related language.
• Experience in algorithms & data structures, parsing, numerical optimization, data mining, parallel/distributed computing, or high‑performance computing.

• Experience building machine‑learning models or developing algorithms 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 to ensure exceptional customer service; and following all federal, state, and local laws and Company policies. Criminal history may have a direct, adverse, or negative relationship with some of the material job duties of this position, such as the abilities to adhere to company policies, exercise sound judgment, manage stress, and protect the business operations and the Company’s reputation.

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

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit (Use the "Apply for this Job" box below). for more information.

Amazon also offers comprehensive benefits, including health insurance (medical, dental, vision, prescription, basic life & AD&D insurance, and an optional supplemental life plan), EAP, mental health support, medical advice line, flexible spending accounts, adoption and surrogacy reimbursement coverage, 401(k) matching, paid time off, and parental leave. Learn more about our benefits at .

USA, CA, Cupertino –  –  USD annually