Software Development Engineer - Testing Infrastructure & Performance, Trainium/Neuron, Annapurna Labs

Benefits

Parental leave

6 weeks From the posting source checked Jun 20, 2026

Non-birth-parent leave

6 weeks From the posting source checked Jun 20, 2026

Family-building benefits

• Fertility benefits: Offered From the posting source checked Jun 20, 2026
• Adoption assistance: Offered From the posting source checked Jun 20, 2026
• Surrogacy assistance: Not verified

Mental health support

Offered From the posting source checked Jun 20, 2026

Relocation assistance

Not verified

Childcare support

Offered From the posting source checked Jun 20, 2026

Learning budget

Not verified

Verification

Source-linked checked Jun 7, 2026

Salary

$127K-$185K From the posting source checked Jun 20, 2026

401(k) match

Reported from DOL Form 5500 industry filing (not employer-specific)

Was this benefit information wrong? Tell us.

Market context

U.S. role benchmark (BLS OEWS)

$116,543 U.S. median for this role

Projected growth (BLS Employment Projections)

+9.8% - Much faster than average

34% above the BLS role benchmark for software engineering aggregate.

Matched to SOC 15-1252 - Software Engineering aggregate by role bucket.

Source: U.S. Bureau of Labor Statistics, OEWS, May 2024 and Employment Projections, 2024-2034.

Role

Role function

Engineering From the posting source checked Jun 20, 2026

Seniority

Mid From the posting source checked Jun 20, 2026

Schedule

Shift type

Not verified

Weekend work

Not verified

Company

Equity

Offered Verified - SEC 10-K source checked Jun 20, 2026

Application

Cover letter

Not verified

Assessment

Not verified

Deadline

Not stated

Where they hire

State eligibility is not yet verified.

About this role

Software Development Engineer - Testing Infrastructure & Performance, Trainium/Neuron, Annapurna Labs Cupertino, California, USA The Neuron team within Annapurna ML develops the software stack that powers AWS's custom AI training and inference chips (Trainium, Inferentia). Our team builds and operates the testing infrastructure and performance characterization systems that ensure every release meets quality and performance bars before reaching customers. We directly enable the velocity and reliability of Neuron software releases. We are looking for a Software Development Engineer to own and evolve the testing infrastructure and performance pipeline that validates Neuron software across workloads, models, and hardware configurations. You will build data pipelines that collect, process, and surface performance metrics - and dashboards that give engineers and leadership real-time visibility into regression detection, performance trends, and release readiness. Your work directly impacts how fast we ship high-quality software to customers running ML workloads on Trainium. Key job responsibilities - Design, build, and maintain automated testing infrastructure (Python) that validates Neuron compiler, runtime, and framework integrations across hardware targets - Build and operate data pipelines that ingest performance benchmarks, test results, and system metrics from distributed test runs into centralized data stores - Develop performance dashboards that surface regression detection, trend analysis, and release-readiness signals to engineering teams and leadership - Create and maintain integration test frameworks that validate end-to-end model compilation, execution, and performance across Trainium configurations - Build automated alerting and triage tooling that identifies performance regressions early and routes them to the right owners - Collaborate with Compiler,

Read the full description at www.amazon.jobs. FewerJobs shows a preview and links to the original posting.