💼 Full-Time Position

Sr. Tech Product Manager - Neuron Runtime and Cloud ML Infrastructure, Annapurna Labs

🏢
Annapurna Labs (U.S.) Inc.
📍 Cupertino, California, United States
📍
Location
Cupertino, United States
📅
Posted
April 12, 2026
Type
Full-Time
🎯

Full-Time Opportunity: This is a permanent, full-time position with a competitive package and real career growth potential.

Job Description

AWS Trainium is deployed at scale, with millions of chips in production, and has been used for training and inference of frontier models. AWS Neuron is the software stack for Trainium, enabling customers to run deep learning and generative AI workloads with optimal performance and cost efficiency.

AWS Neuron is hiring a Technical Product Manager to work backward from Trainium customers and drive the developer experience for running high-performance ML workloads at scale on AWS Trainium, from getting started with Neuron Deep Learning Containers, AMIs, and AWS services to operating at scale through orchestration, resiliency, and observability.

You will drive the product strategy for how developers interact with Trainium through container ecosystems, resource management platforms, and AWS services. This includes Neuron integration with orchestration tools (SLURM, Kubernetes), AWS services (EKS, SageMaker), Neuron Deep Learning Containers and AMIs, and Linux distribution ...