jobs

Machine Learning - Compiler Engineer , AWS Neuron, Annapurna Labs

Do you want to be part of AI revolution? At Amazon our vision is to make deep learning pervasive for everyday developers and to democratize access to AI hardware and software infrastructure. In order to deliver on that vision, we’ve created

In inglese

Company

Amazon

Location

Cupertino, California, USA

Status

Open

Posted

2026-09-17T00:00:00+00:00

Amazon's Annapurna Labs is hiring a software engineer for the Compiler team behind AWS Neuron, the SDK that runs ML models on Inferentia and Trainium chips. The work covers building the next-generation Neuron compiler, which converts models written in PyTorch, TensorFlow and JAX, plus writing optimization and verification passes and fixing compiler defects. The posting asks for three years of professional software development, two in design or architecture, and C++ or Java. Compiler or accelerator experience is preferred, not required; open-source work on MLIR, OpenXLA and StableHLO is a bonus.