Drimva

Amazon

Sr Embedded Software Engineer, Annapurna ML Software

$168K–$262K

Austin, TX +1 · Senior · Full-time · Posted Sep 28, 2026

Location
Austin, TX
Cupertino, CA
Experience
5+ years for each of the four experience areas listed in the basic qualifications.
Education
Bachelor's degree in computer science or equivalent.

About the job

Architects and develops software and firmware for NeuronSwitches, bringing up and qualifying hardware and optimizing the high-performance interconnect fabric for Trainium systems.

What you'll do

  • Lead architecture and technical strategy for the machine-learning platform; develop software and firmware, including device drivers and switch-fabric control logic.
  • Collaborate with EC2, Annapurna Labs, and manufacturing teams on system integration, hardware bring-up, and board-level debugging.
  • Own qualification frameworks and processes, and develop tooling for testing and production deployment.

What they ask for

  • 5+ years of non-internship professional software development experience.
  • 5+ years programming with at least one software programming language.
  • 5+ years leading design or architecture of new and existing systems, including design patterns, reliability, and scaling.
  • 5+ years of full software development life cycle experience; experience as a mentor, tech lead, or engineering-team lead.

Nice to have

  • Bachelor's degree in computer science or equivalent.

Pay

193,300.00 - 261,500.00 USD annually (base salary); 168,100.00 - 227,400.00 USD annually (base salary)

Tools

C/C++, Linux, NeuronSwitches, Trainium 3, EC2

AI focus

Develops software and firmware for the switching infrastructure that supports machine-learning workloads and Trainium chip-to-chip communication.

Model-written summary · the original posting is authoritative

First seen Oct 15, 2026 · Last verified Nov 4, 2026 (UTC)