Job Description
Description
Our team is responsible for the AWS Neuron Compiler, a cutting-edge deep learning compiler stack that powers Generative AI and other advanced ML workloads on AWS’s custom-built ML accelerators — Inferentia and Trainium. These accelerators deliver best-in-class performance and cost-efficiency for ML inference and training in the cloud.
We’re building a new core group of engineers in TLV to drive innovation in compiler technology and systems-level ML software. As a Machine Learning Compiler Engineer, you’ll help shape the direction of the team from the ground up and work on:
Designing and optimizing compiler features for high-performance ML workloads
Solving complex challenges in areas like graph optimization, code generation, and hardware-software co-design
Collaborating across Runtime, Frameworks, and Hardware teams to maximize end-to-end performance
As part of the Backend Compiler Team, you’ll contribute to projects involving...
Our team is responsible for the AWS Neuron Compiler, a cutting-edge deep learning compiler stack that powers Generative AI and other advanced ML workloads on AWS’s custom-built ML accelerators — Inferentia and Trainium. These accelerators deliver best-in-class performance and cost-efficiency for ML inference and training in the cloud.
We’re building a new core group of engineers in TLV to drive innovation in compiler technology and systems-level ML software. As a Machine Learning Compiler Engineer, you’ll help shape the direction of the team from the ground up and work on:
Designing and optimizing compiler features for high-performance ML workloads
Solving complex challenges in areas like graph optimization, code generation, and hardware-software co-design
Collaborating across Runtime, Frameworks, and Hardware teams to maximize end-to-end performance
As part of the Backend Compiler Team, you’ll contribute to projects involving...
Apply for this Position
Ready to join Amazon? Click the button below to submit your application.
Submit Application