Compiler Engineer — AI Model Acceleration (Neuron)

Amazon Inc.; Seattle (WA); On-site; USD 129,000 - 148,000; Full time

Job summary

Amazon Inc. in Seattle/Cupertino seeks a software engineer for the AWS Neuron Compiler team. You will design, implement, test, and deploy compiler optimizations to transform ML models from PyTorch, TensorFlow, and JAX into efficient code for Inferentia/Trainium.

Collaborate with chip architects, runtime engineers, scientists and ML Apps teams to deliver high‑performance, cost-efficient accelerators, with opportunities to contribute to open‑source projects like StableHLO and MLIR.

Qualifications

• 1+ years of non‑internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience.
• Experience programming with at least one software programming language.

Responsibilities

• Design, implement, test, deploy and maintain innovative software solutions to transform Neuron compiler’s performance, stability and user‑interface.
• Work with chip architects, runtime/OS engineers, scientists and ML Apps teams to deploy ML models on AWS accelerators with cost/performance benefits.
• Collaborate on open‑source projects (StableHLO, OpenXLA, MLIR) to optimize ML workloads on AWS software and hardware.
• Build features delivering the best experiences for developers worldwide.

Skills

Software design

Programming languages

Education

Bachelor's degree in computer science

Job description

Amazon Inc. in Seattle/Cupertino seeks a software engineer for the AWS Neuron Compiler team. You will design, implement, test, and deploy compiler optimizations to transform ML models from PyTorch, TensorFlow, and JAX into efficient code for Inferentia/Trainium.

Collaborate with chip architects, runtime engineers, scientists and ML Apps teams to deliver high‑performance, cost-efficient accelerators, with opportunities to contribute to open‑source projects like StableHLO and MLIR.