Senior ML Compiler Engineer — Flexible, High-Impact AI

Amazon Web Services (AWS); Cupertino (CA); On-site; USD 180,000 - 250,000; Full time

Job summary

AWS Neuron compiler team is seeking a senior software engineer to build the next generation compiler that transforms ML models into executable code for AWS Inferentia and Trainium. You will collaborate with chip architects, runtime/OS engineers, and ML Apps teams to deliver high-performance, cost-efficient software, contribute to open-source ecosystems, and ship features for developers worldwide.

You will also engage with external open-source communities and participate in pre-silicon design,

Qualifications

• Bachelor's degree in computer science or equivalent.
• 5+ years of full software development life cycle experience.
• 3+ years of developing compiler features and optimizations.

Responsibilities

• Design, implement, test, deploy and maintain Neuron compiler software.
• Collaborate with chip architects, runtime/OS teams and ML Apps groups.
• Contribute to open-source ecosystems (StableHLO, OpenXLA, MLIR) and pre-silicon design.

Skills

C++

Java

ML frameworks experience

Education

Bachelor's degree in computer science or equivalent

Tools

OpenXLA

StableHLO

MLIR

Job description

AWS Neuron compiler team is seeking a senior software engineer to build the next generation compiler that transforms ML models into executable code for AWS Inferentia and Trainium. You will collaborate with chip architects, runtime/OS engineers, and ML Apps teams to deliver high-performance, cost-efficient software, contribute to open-source ecosystems, and ship features for developers worldwide.

You will also engage with external open-source communities and participate in pre-silicon design,