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 ...
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 ...
Apply for this Position
Ready to join Amazon? Click the button below to submit your application.
Submit Application