Software Engineer, Neuron Compiler ML Accelerators
Listed on 2026-10-18
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer, Software Engineer
Amazon Inc. in Los Angeles/California area is seeking a Software Development Engineer for the AWS Neuron compiler team. You will help transform ML models from frameworks like PyTorch, Tensor Flow, and JAX to run efficiently on AWS Inferentia and Trainium chips in the cloud.
The role emphasizes compiler optimization, collaboration with hardware and software teams, and involvement with open-source tool chains such as StableHLO, OpenXLA, and MLIR. Prior OO language experience is required.
This position is for the Software Engineer, Neuron Compiler for ML Accelerators role at Amazon Inc.
Step into the Software Engineer, Neuron Compiler for ML Accelerators role at Amazon Inc. in Cupertino, CA, United States and grow with us.
Please review the full job details above before applying.
If your experience matches this role, we encourage you to apply.
(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).