πŸ‡ΊπŸ‡Έ USAJobs.work

America's Job Portal

← Back to USA Jobs

Software Engineer Neuron Containers, Annapurna Labs, Neuron Containers, Annapurna Labs

Company

Amazon

Location

Cupertino, CA

Posted

July 18, 2026

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...

Ready to Apply?

Join thousands of Americans building their careers

Apply Now