ML Compiler Engineer AI Inference Accelerators
Listed on 2026-10-06
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer
Amazon is seeking a software engineer for the AWS Neuron Compiler team in Cupertino, CA. You will build the next‑generation Neuron compiler to translate ML models from PyTorch, Tensor Flow, and JAX for deployment on AWS Inferentia/Trainium in the cloud.
You will work with OpenXLA, StableHLO and MLIR, design optimization passes, and collaborate with chip architects, runtime engineers, scientists, and ML apps teams to deliver high performance and a great developer experience.
Consider building your career as a ML Compiler Engineer for AI Inference Accelerators at Amazon Inc.
Learn more about the ML Compiler Engineer for AI Inference Accelerators role in the description above.
We appreciate your interest in this position.
Join Amazon Inc. and contribute to our ongoing work.
Take a moment to read everything above and see whether this role is right for you.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).