Software Engineer II: AI/ML Inference Custom Hardware
Listed on 2026-10-07
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer
Amazon Web Services (AWS) Annapurna Labs team is hiring a Software Engineer II - AI/ML to design and implement features and own model performance on AWS Trainium end to end. You will work across ML frameworks, kernels, and runtimes to optimize inference and enable scale for GenAI workloads.
You'll collaborate with customers to enable and optimize their models on AWS accelerators, build high-performance kernels, and contribute to compiler/runtime improvements across multiple generations of Neuron
The Software Engineer II: AI/ML Inference on Custom Hardware position in the IT & Technology field is open for applications.
Take a moment to read everything above and see whether this role is right for you.
This posting is for the Software Engineer II: AI/ML Inference on Custom Hardware role at Amazon Inc., based in Cupertino, CA, United States.
(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).