Open jobs in the United States

Save this search

Added 10 hours ago
Applied Scientist, Neuron ARG
Annapurna Labs (U.S.) Inc.
Seattle, WA
Added 1 day ago
Sr. Systems Development Engineer, Trainium Systems
Annapurna Labs (U.S.) Inc.
Austin, TX · Seattle, WA · San Diego, CA · Cupertino, CA
Added 2 days ago
Sr. Software Development Engineer, ML Infrastructure Team
Annapurna Labs (U.S.) Inc.
Cupertino, CA
Added 2 days ago
Product Development Engineer, Annapurna Labs Silicon Operations
Annapurna Labs (U.S.) Inc.
Austin, TX
Added 5 days ago
Software Development Engineer, Neuron Collectives, Annapurna Labs
Annapurna Labs (U.S.) Inc.
Cupertino, CA
Added 5 days ago
Thermal & Mechanical Packaging Engineer, Annapurna Labs
Annapurna Labs (U.S.) Inc.
Austin, TX
Added 5 days ago
ASIC Formal Verification Engineer I, Annapurna Labs
Annapurna Labs (U.S.) Inc.
Austin, TX · Cupertino, CA
Added 6 days ago
Package Engineer, Annapurna Labs - AI Silicon Packaging
Annapurna Labs (U.S.) Inc.
Tempe, AZ · Austin, TX
Added 6 days ago
Sr. Manager, Foundation IP (Standard Cells & Memories),, Annapurna Labs
Annapurna Labs (U.S.) Inc.
Austin, TX · Cupertino, CA
Added 6 days ago
Sr. Physical Design Verification Engineer, Annapurna Labs
Annapurna Labs (U.S.) Inc.
Austin, TX · Cupertino, CA
Added 7 days ago
Sr. Package Engineer, Annapurna Labs - AI Silicon Packaging
Annapurna Labs (U.S.) Inc.
Tempe, AZ · Austin, TX
Added 7 days ago
Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training
Annapurna Labs (U.S.) Inc.
Seattle, WA
Added 8 days ago
Sr. Hardware Engineer - ML Acceleration, Annapurna Labs
Annapurna Labs (U.S.) Inc.
Austin, TX · Cupertino, CA
Added 8 days ago
Software Development Engineer, Annapurna Labs, Elastic Collectives
Annapurna Labs (U.S.) Inc.
Cupertino, CA
Added 8 days ago
Sr. PD Methodology Engineer, Annapurna Labs - Cloud Scale Machine Learning
Annapurna Labs (U.S.) Inc.
Austin, TX · Cupertino, CA
Added 8 days ago
Sr. PD Methodology Engineer, Annapurna Labs - Cloud Scale Machine Learning
Annapurna Labs (U.S.) Inc.
Austin, TX · Cupertino, CA
Added 8 days ago
Applied Scientist, Neuron ARG, Annapurna ML
Annapurna Labs (U.S.) Inc.
Seattle, WA · Cupertino, CA
Added 11 days ago
ML Compiler Engineer, Neuron Kernel Interface
Annapurna Labs (U.S.) Inc.
Seattle, WA · Cupertino, CA
Added 11 days ago
Senior Product Development Engineer, Annapurna Labs Silicon Operations
Annapurna Labs (U.S.) Inc.
Austin, TX
Added 12 days ago
Principal Technical Program Manager, Annapurna Labs, Inc
Annapurna Labs (U.S.) Inc.
Austin, TX · Cupertino, CA
Added 13 days ago
Software Development Engineer, AWS Neuron, Annapurna Labs
Annapurna Labs (U.S.) Inc.
Cupertino, CA
Added 13 days ago
Software Development Engineer III, Machine Learning Annapurna - System Integration
Annapurna Labs (U.S.) Inc.
Seattle, WA · Cupertino, CA
Added 13 days ago
Principal Engineer: Annapurna Labs Training and Inference , Annapurna Labs, AWS Neuron SDK
Annapurna Labs (U.S.) Inc.
Seattle, WA · Cupertino, CA
Added 14 days ago
Executive Assistant, Annapurna Labs
Annapurna Labs (U.S.) Inc.
Seattle, WA
Added 14 days ago
Software Engineer II - AI/ML, Neuron Inference, Inference
Annapurna Labs (U.S.) Inc.
Cupertino, CA

Annapurna Labs (U.S.) Inc.

Runtime/Driver Software Development Engineer, Neuron Runtime

Austin, TX, United States

Apply About employer

Applications open in a new tab.

Job details

Posting language: English
Posted: Jun 12, 2026

Original job description

Neuron SDK is the complete software stack for the AWS Inferentia and Trainium machine learning accelerators designed by Annapurna Labs inside AWS. The Neuron SDK consists of a compiler, runtime, frameworks, and tooling customers need. It’s also preinstalled in Amazon Deep Learning AMIs and Deep Learning Containers for customers to quickly get started with running high performance and cost-effective inference and training.

This position is for a Software Engineer for the AWS Neuron SDK team with a deep background in Linux and device drivers. Deep knowledge and experience with hardware-based accelerator platforms and the necessary complexity for the software layers that interact and manage them. Experience implementing high performance software drivers, needed to ensure high throughput from software to hardware, is also a necessity. This engineer will work with the team to define the key interfaces and will develop the runtime stack and driver that manages the accelerator. This engineer will also work in pre-silicon environments (such as emulation) to shift-left Runtime Software Stack readiness and to develop validation content to checkout HW/Silicon features.

This is a fast-paced, intellectually challenging position, where you’ll work with thought-leaders in multiple technology areas. You’ll have high standards for yourself and everyone you work with, and you’ll be constantly looking for ways to improve our products' performance, quality and cost.

We’re changing an industry, and we want individuals who are ready for this challenge and want to reach beyond what is possible today.

Basic 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 Preferred qualifications
3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
Bachelor's degree in computer science or equivalent
Experience in computer architecture
Experience writing low level drivers

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits.

USA, TX, Austin - 143,700.00 - 194,400.00 USD annually

Apply About employer

More about this job

Skills

Software development
AWS
Linux
Computer science
Recruitment
Mental health

Report this job