AI/ML Systems Engineer – Distributed Training Trainium
Listed on 2026-10-06
-
IT/Tech
Systems Engineer, Machine Learning/ ML Engineer, AI Engineer (Applied/Software)
Annapurna Labs (U.S.) Inc. within AWS is seeking engineers to optimize distributed training on Trainium. You will work across PyTorch and Neuron stack with compiler and runtime teams to improve training throughput on AWS accelerators.
The role focuses on defining parallelism strategies, profiling workloads, and translating performance gaps into framework requirements, contributing upstream to open-source projects.
Are you ready to take on the AI/ML Systems Engineer – Distributed Training on Trainium role at Amazon Web Services (AWS)?
We have an opening for a AI/ML Systems Engineer – Distributed Training on Trainium in Cupertino, CA, United States within IT & Technology.
This role, AI/ML Systems Engineer – Distributed Training on Trainium at Amazon Web Services (AWS), could be your next career step.
We would love to welcome a new AI/ML Systems Engineer – Distributed Training on Trainium to our team 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).