AI/ML Software Engineer - Distributed Training HPC
Listed on 2026-10-17
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer, Software Engineer
Amazon’s Annapurna Labs team within AWS Neuron Training seeks a Software Engineer to optimize distributed training on Trainium, tackling throughput across the Neuron stack and collaborating with PyTorch and JAX integrations. You will own parallelism strategies and help land improvements across compiler, runtime, and collectives teams.
The role emphasizes performance tuning, end-to-end profiling, and upstream contributions to open source frameworks used by customers training frontier models on
Join us at Amazon as our next AI/ML Software Engineer
- Distributed Training HPC in Cupertino, CA, United States.
For the AI/ML Software Engineer
- Distributed Training HPC position at Amazon, we are reviewing applications now.
Step into the AI/ML Software Engineer
- Distributed Training HPC role at Amazon in Cupertino, CA, United States and grow with us.
Please review the full job details above before applying.
(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).