Senior AI/ML Distributed Training Engineer
Listed on 2026-10-07
-
Software Development
Machine Learning/ ML Engineer, AI Engineer (Applied/Software)
Amazon Annapurna Labs is seeking a Sr. Software Engineer for AI/ML distributed training to design and optimize large-scale ML training on Trainium instances. You will enhance distributed frameworks and optimize mixed-precision training, working closely with hardware and runtime teams.
Qualified candidates have 5+ years in software development, leadership experience, and strong PyTorch expertise. On-site in Cupertino, you will collaborate with AWS solution architects and customers to deploy
For the Senior AI/ML Distributed Training Engineer position at Amazon, we are reviewing applications now.
Join Amazon and contribute to our ongoing work.
Take a moment to read everything above and see whether this role is right for you.
This posting is for the Senior AI/ML Distributed Training Engineer role at Amazon, based in Cupertino, CA, United States.
We are looking to fill the Senior AI/ML Distributed Training Engineer position at Amazon in Cupertino, CA, United States.
(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).