← All roles

Machine Learning - Compiler Engineer

Amazon· AWS Neuron, Annapurna Labs
Apply on company site
Cupertino$165.2k - $223.6kFull-timePosted 2w ago

About the role

Software engineer in the Compiler team for AWS Neuron, building next-generation Neuron compiler to transform ML models for AWS Inferentia and Trainium, collaborating with chip architects, runtime/OS engineers, scientists, and ML Apps teams to optimize performance and usability.

Responsibilities

  • Design, implement, test, deploy and maintain 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 optimal cost/performance benefits.
  • Collaborate with open-source software (e.g., StableHLO, OpenXLA, MLIR) to optimize ML workloads on AWS software/hardware.
  • Build features that deliver best possible experiences for developers globally.
  • Create compiler optimization and verification passes, surface features/peculiarities of AWS accelerators to developers, analyze numerical errors, and resolve compiler defects.
  • Participate in design discussions, code reviews, and communicate with internal and external stakeholders.

Qualifications

  • 3+ years of non-internship professional software development experience
  • 2+ years of non-internship design or architecture experience (design patterns, reliability and scaling) of new and existing systems
  • Experience programming with at least one software programming language (C++/Java required)

Skills

Required:Software developmentObject-oriented programming
Nice to have:Compiler design / optimizationOpenSource toolchainsDeep learning frameworks

Benefits

  • Base salary + sign-on payments and RSUs
  • Comprehensive benefits including health insurance (medical, dental, vision, prescription, life & AD&D, optional life), EAP, Mental Health Support, FSA, Adoption and Surrogacy Reimbursement
  • 401(k) matching, paid time off, parental leave
Last updated Oct 25, 2026. Interested? Apply on the company site.

About Amazon

Amazon.com, Inc. or its affiliates

liquid-ai-recruiting-target
Company profileWebsite

More open roles at Amazon