AI/ML Systems Engineer – Distributed Training Trainium
Listed on 2026-11-04
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer
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.
We are currently recruiting a AI/ML Systems Engineer – Distributed Training on Trainium for our team in Cupertino, CA, United States.
Our team is growing, and we are hiring a AI/ML Systems Engineer – Distributed Training on Trainium in Cupertino, CA, United States.
This opportunity is for the AI/ML Systems Engineer – Distributed Training on Trainium role at Amazon Web Services (AWS).
We are seeking a motivated AI/ML Systems Engineer – Distributed Training on Trainium to join Amazon Web Services (AWS) 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).