Hardware-Aware ML Scientist II: Trainium & Inferentia
Listed on 2026-10-20
-
IT/Tech
Machine Learning/ ML Engineer, AI Engineer (Applied/Software), Data Scientist
Amazon seeks an Applied Scientist II to push hardware-aware ML for Trainium and Inferentia. You will own model and algorithm decisions from research through production, publish at top venues, and work alongside distinguished engineers and scientists in a strategic AWS growth area.
We actively work on low-precision training and inference, Trn4 datatype exploration, and system-aware optimizers. Develop production-quality PyTorch or JAX code and mentor others, shaping the technical roadmap.
The following opening is for a Hardware-Aware ML Scientist II:
Trainium & Inferentia with Amazon Inc.
Please review the full job details above before applying.
If your experience matches this role, we encourage you to apply.
All applications are reviewed carefully by our team.
The position is based in Seattle, WA, United States.
This opportunity is part of our work in IT & Technology, Other.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).