ML Compiler Engineer for Neuron – DL on Inferentia

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

Job summary

Amazon in Seattle and Cupertino is hiring a software engineer for the AWS Neuron compiler team. You will design, implement, test, deploy and maintain compiler passes that transform ML models to run efficiently on AWS Inferentia and Trainium hardware.

You will collaborate with chip architects, runtime engineers, scientists, and ML Apps teams to deliver state-of-the-art performance, explore open-source tools like StableHLO, OpenXLA, MLIR, and help bring new features to market.

Qualifications

• 1+ years of non-internship design or architecture of new and existing systems.
• Experience programming with at least one software programming language.
• 3+ years of full software development life cycle, including coding standards, code reviews, source control, build processes, testing, and operations.
• Bachelor's degree in computer science or equivalent.

Responsibilities

• Design, implement, test, deploy and maintain compiler passes for Neuron compiler.
• Collaborate with chip architects, runtime/OS engineers, scientists and ML Apps teams.
• Explore and integrate open-source tools like StableHLO, OpenXLA, MLIR.
• Deliver features that optimize ML workloads on AWS accelerators with cost/performance benefits.

Skills

Programming languages

SDLC experience

Education

Bachelor's degree in computer science or equivalent

Tools

OpenXLA

StableHLO

MLIR

Job description

Amazon in Seattle and Cupertino is hiring a software engineer for the AWS Neuron compiler team. You will design, implement, test, deploy and maintain compiler passes that transform ML models to run efficiently on AWS Inferentia and Trainium hardware.

You will collaborate with chip architects, runtime engineers, scientists, and ML Apps teams to deliver state-of-the-art performance, explore open-source tools like StableHLO, OpenXLA, MLIR, and help bring new features to market.