Neuron Runtime Software Engineer — Flexible Hours

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

Benefits offered by this job

Health insurance

401(k) matching

Paid time off

Parental leave

RSUs

Job summary

Amazon's Annapurna Labs (U.S.) Inc. in Cupertino is seeking a Software Development Engineer for the Neuron Runtime Team to build and maintain high‑performance runtime libraries and drivers for ML accelerators and servers.

You will collaborate across teams to ensure the C++ compiler exposes critical performance insights for devices like Trainium and Inferentia and drive improvements to support PyTorch, JAX, and XLA frameworks.

Qualifications

• 3+ years of non‑internship professional software development experience.
• 2+ 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

• Develop and maintain high‑performance runtime libraries and drivers for ML accelerators and servers.
• Drive design, development and deployment of Neuron Runtime and other Neuron components.
• Collaborate across teams to ensure the C++ compiler exposes key information for performance optimization.

Skills

Software development

System design

Programming

Education

Bachelor's degree in computer science

Job description

Amazon's Annapurna Labs (U.S.) Inc. in Cupertino is seeking a Software Development Engineer for the Neuron Runtime Team to build and maintain high‑performance runtime libraries and drivers for ML accelerators and servers.

You will collaborate across teams to ensure the C++ compiler exposes critical performance insights for devices like Trainium and Inferentia and drive improvements to support PyTorch, JAX, and XLA frameworks.