Linux Driver & Runtime Engineer for ML Accelerator SDK

Amazon Web Services (AWS); Austin (TX); On-site; USD 143,700 - 194,400; Full time

Job summary

Annapurna Labs (U.S.) Inc. is seeking a Software Engineer for the AWS Neuron SDK team in Austin, TX. You will build the runtime stack and device drivers to maximize performance on AWS Inferentia and Trainium accelerators.

The role emphasizes Linux driver development, low-level software, pre-silicon validation, and interfaces with hardware teams to push throughput, reliability, and cost efficiencies across inference and training workloads.

Qualifications

• 3+ years of professional software development experience.
• 2+ years of design or architecture experience for new or existing systems.
• Experience programming in at least one software language.

Responsibilities

• Define interfaces and develop runtime stack and driver for accelerators.
• Work in pre-silicon environments (emulation) to shift-left readiness and validation content.
• Collaborate with hardware and software teams to maximize throughput and cost efficiency.
• Deliver high-quality, scalable software in a fast-paced environment.

Skills

Software development

System design

Programming languages

Education

Bachelor's degree in Computer Science or equivalent

Tools

Low level drivers

Job description

Annapurna Labs (U.S.) Inc. is seeking a Software Engineer for the AWS Neuron SDK team in Austin, TX. You will build the runtime stack and device drivers to maximize performance on AWS Inferentia and Trainium accelerators.

The role emphasizes Linux driver development, low-level software, pre-silicon validation, and interfaces with hardware teams to push throughput, reliability, and cost efficiencies across inference and training workloads.