Senior Applied Scientist, Hardware-Aware ML for Trainium

Amazon Science; Cupertino (CA); On-site; USD 192,000 - 260,000; Full time

Benefits offered by this job

Health insurance

401(k) matching

Paid time off

Parental leave

Job summary

Amazon Science is seeking an Applied Scientist to push the frontier of hardware-aware ML for Trainium and Inferentia. This role combines research with production impact, owning solutions from concept to deployment and publishing in top venues.

You will collaborate with silicon teams, mentor others, and help shape the technical roadmap for AWS ML accelerators. The position emphasizes building scalable ML systems, applying advanced deep learning methods, and delivering high-quality,

Qualifications

• PhD, or Master's degree with 6+ years of applied research experience.
• Experience programming in Java, C++, Python.
• 3+ years of building machine learning models for business application experience.
• Experience with neural deep learning methods and machine learning.

Responsibilities

• Own scientific problems end-to-end - from research and experimentation through production impact - applying rigorous evaluation to complex, ill-defined problems at large scale.
• Develop production-quality code in PyTorch or JAX and integrate scientific components into large-scale training and inference systems with operational excellence and efficient resource usage.
• Partner with foundation-model, engineering, and hardware-architecture teams so your findings directly inform what gets built into Trainium and shipped in the product stack.
• Mentor fellow scientists and interns, give constructive peer reviews, and help shape team goals, priorities, and the technical roadmap.
• Author and publish research at top peer-reviewed venues (ICLR, NeurIPS, ICML, MLSys) and engage the broader scientific community.

Skills

PhD or MS + 6+ yrs

Java

C++

Python

ML model development

Deep learning experience

Education

PhD

Master's degree

Tools

PyTorch

JAX

Job description

Amazon Science is seeking an Applied Scientist to push the frontier of hardware-aware ML for Trainium and Inferentia. This role combines research with production impact, owning solutions from concept to deployment and publishing in top venues.

You will collaborate with silicon teams, mentor others, and help shape the technical roadmap for AWS ML accelerators. The position emphasizes building scalable ML systems, applying advanced deep learning methods, and delivering high-quality,