ML Compiler Engineer Custom ML Accelerators
Listed on 2026-10-07
-
Software Development
Machine Learning/ ML Engineer, AI Engineer (Applied/Software)
Amazon Web Services (AWS) is seeking a skilled ML compiler engineer to help build a state-of-the-art deep learning compiler stack that optimizes models for PyTorch, Tensor Flow, and JAX across accelerators like Inferentia/Trainium.
You will design, implement, and optimize compiler features, collaborate with Runtime, Frameworks, and Hardware teams, and contribute to the Neural Compiler effort enabling scalable AI in the cloud.
We have an opening for a ML Compiler Engineer for Custom ML Accelerators in Seattle, WA, United States within IT & Technology, Engineering.
Join us at Amazon Web Services (AWS) as our next ML Compiler Engineer for Custom ML Accelerators in Seattle, WA, United States.
We are currently recruiting a ML Compiler Engineer for Custom ML Accelerators for our team in Seattle, WA, United States.
This is a strong role to take on the ML Compiler Engineer for Custom ML Accelerators role at Amazon Web Services (AWS).
(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).