← All roles

Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs

Amazon· Annapurna Labs (U.S.) Inc.
Apply on company site
Cupertino$193.3k - $261.5kFull-timePosted 2w ago

About the role

Senior software engineer in the Compiler team for AWS Neuron, responsible for building and optimizing the Neuron compiler to transform ML models for AWS Inferentia/Trainium accelerators, interfacing with open-source ecosystems, and collaborating across hardware and software teams.

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 optimal cost/performance benefits.
  • Engage with open-source communities (StableHLO, OpenXLA, MLIR) and influence industry partners to optimize ML workloads on AWS software and hardware.
  • Build features to deliver the best experiences for developers worldwide.

Qualifications

  • 5+ years of non-internship professional software development experience
  • 5+ years of programming with at least one software programming language experience
  • 5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience
  • Experience as a mentor, tech lead or leading an engineering team

Skills

Required:C++Compiler design / optimizationLeadership / mentoring
Nice to have:JavaML frameworks on accelerators (e.g., GPUs)

Benefits

  • Health insurance (medical, dental, vision, prescription)
  • Basic Life & AD&D insurance and option for Supplemental life plans
  • Employee Assistance Program (EAP), Mental Health Support
  • Flexible Spending Accounts
  • Adoption and Surrogacy Reimbursement coverage
  • 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