Senior AI/ML Systems Engineer - Distributed Training
Listed on 2026-10-06
-
Software Development
Machine Learning/ ML Engineer, AI Engineer (Applied/Software)
Amazon Neuron Training in Cupertino seeks a Sr. Software Engineer to lead distributed training efforts for Trainium accelerators.
You will work across PyTorch and JAX to enable large-scale training, post-training, and RL workflows, owning parallelism strategies and driving performance improvements with compiler, runtime, and framework teams. Ideal candidates bring 5+ years of software leadership, deep ML experience, and a strong background in distributed systems.
As a Senior AI/ML Systems Engineer
- Distributed Training, you will play an important part at Amazon in Cupertino, CA, United States.
Take a moment to read everything above and see whether this role is right for you.
This posting is for the Senior AI/ML Systems Engineer
- Distributed Training role at Amazon, based in Cupertino, CA, United States.
We are looking to fill the Senior AI/ML Systems Engineer
- Distributed Training 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).