Position Overview
Description
Annapurna Labs was a startup acquired by AWS in 2015 and is now fully integrated. AWS Neuron is
the complete software stack for the Inferentia and Trainium ML accelerators β the custom silicon that powers large-scale AI workloads on AWS.
The Neuron Containers team is looking for a Software Development Engineer to build platform integrations that enable customers to run distributed training and inference workloads on
Neuron at scale. The team owns Neuron integration with Kubernetes, ECS, and Slurm β handling device allocation, fault tolerance, auto-scaling, and orchestration across large clusters. The team also owns delivery of Neuron Deep Learning Containers (DLCs) and Deep Learning AMIs
(DLAMIs) β the pre-configured container images and machine images that package the Neuron SDK
for customer deployment on Trainium and Inferentia instances.
Key job responsibilities
Design and implement container platform integrations β device plugins, DRA...