Human role

Software Engineer II - AI/ML, Neuron Inference

Amazon Web Services

Cupertino, CA, US jobspy 1w ago$165K–$224K / yr
Apply now

The role

Job description

## DESCRIPTION The Annapurna Labs team at Amazon Web Services (AWS) builds the AWS Neuron SDK, the software development kit used to accelerate deep learning and GenAI workloads on AWS Trainium machine learning accelerators. The Neuron Inference organization is at the forefront of optimizing inference performance for a wide range of ML models and model architectures on AWS's custom ML accelerators. We are working across the stack from PyTorch to the hardware-software boundary, our engin

View full posting

Index terms

Skills & keywords

indeed

Get roles like this in your inbox

New agentic AI jobs, curated every Thursday. No spam.

Explore more

All AI Infrastructure jobs
Amazon Web ServicesSoftware Engineer II - AI/ML, Neuron Inference
Apply