ML Systems Engineer — AI/GenAI Training & HPC
Job in
Cupertino, Santa Clara County, California, 95014, USA
Listed on 2026-08-19
Listing for:
Amazon Web Services (AWS)
Apprenticeship/Internship
position Listed on 2026-08-19
Job specializations:
-
IT/Tech
Machine Learning/ ML Engineer, AI Engineer (Applied/Software)
Job Description & How to Apply Below
Annapurna Labs at AWS (Amazon Web Services) builds AWS Neuron to accelerate deep learning and GenAI workloads on Trainium. The team works across ML compiler, runtime, collectives, and framework integration with PyTorch/JAX to train frontier models.
You will lead efforts to optimize distributed training throughput on Trainium, collaborating with PyTorch and Neuron teams to enable large-scale training. The role blends ML, high-performance computing, and distributed systems in a fast-paced
#J-18808-LjbffrTo View & Apply for jobs on this site that accept applications from your location or country, tap the button below to make a Search.
(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).
(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).
Search for further Jobs Here:
×