Applied Scientist II: Hardware-Aware ML for Trainium

Amazon Inc.; Cupertino (CA); Hybride; USD 172 000 - 222 000; Plein temps

Résumé du poste

Amazon's AWS Neuron Science Core Algorithm team seeks an Applied Scientist II to push hardware-aware ML for Trainium and Inferentia. You own model and algorithm decisions from research through production, publish at top venues, and collaborate with engineers and scientists in a strategic growth area for AWS.

You will develop production-quality PyTorch or JAX code, integrate scientific components into large-scale training and inference systems, and mentor fellow scientists while contributing to

Qualifications

• 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.
• 3+ years of building models for business application experience.
• Experience programming in Java, C++, Python or related language.
• Experience in state-of-the-art deep learning models architecture design and deep learning training and optimization and model pruning.

Responsabilités

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

Connaissances

Java

C++

Python

Deep learning

Model architecture

Research publications

Patent experience

Formation

PhD or Master's + 4+ years

Outils

PyTorch

JAX

Description du poste

Amazon's AWS Neuron Science Core Algorithm team seeks an Applied Scientist II to push hardware-aware ML for Trainium and Inferentia. You own model and algorithm decisions from research through production, publish at top venues, and collaborate with engineers and scientists in a strategic growth area for AWS.

You will develop production-quality PyTorch or JAX code, integrate scientific components into large-scale training and inference systems, and mentor fellow scientists while contributing to