Senior AI/ML Systems Engineer — Distributed Training
Listed on 2026-10-20
-
Software Development
Machine Learning/ ML Engineer, AI Engineer (Applied/Software)
Amazon in Cupertino, CA is seeking a Senior Software Engineer- AI/ML to help build and optimize distributed training for AWS Neuron on Trainium. You will work across PyTorch and JAX, shaping parallelism strategies and high-performance kernels, collaborating with compiler, runtime, and open source frameworks.
The role blends ML, HPC, and distributed systems, with mentoring responsibilities and opportunities to influence future accelerator architecture.
This posting is for the Senior AI/ML Systems Engineer — Distributed Training role at Amazon Inc., based in Cupertino, CA, United States.
All applications are reviewed carefully by our team.
The position is based in Cupertino, CA, United States.
This opportunity is part of our work in IT & Technology.
The advertised compensation is 193..
We aim to respond to suitable candidates as soon as possible.
(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).