Drimva

Amazon

Neuron Runtime Software Development Engineer , Neuron Runtime

$144K–$194K

Seattle, WA · Full-time · Posted Apr 10, 2026

Experience
3+ years of non-internship professional software development; 2+ years of non-internship systems design or architecture experience. Preferred: 3+ years of full software development life cycle experience.
Education
Preferred: Bachelor's degree in computer science or equivalent.

About the job

Develop and maintain high-performance Neuron Runtime libraries, drivers, and profiling capabilities for machine-learning applications and AWS Inferentia and Trainium accelerators.

What you'll do

  • Design, develop, deploy, and operate Neuron Runtime and related components across their full development life cycle.
  • Improve ML kernel and framework performance and build profiler capabilities that help customers analyze workloads and hardware behavior.
  • Collaborate across teams on compiler-generated performance information and support profiling across frameworks including PyTorch, JAX, and XLA.

What they ask for

  • 3+ years of non-internship professional software development experience.
  • 2+ years of non-internship experience in design or architecture of new and existing systems, including design patterns, reliability, and scaling.
  • Experience programming with at least one software programming language.

Nice to have

  • 3+ years of full software development life cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
  • Bachelor's degree in computer science or equivalent.

Pay

143,700.00 - 194,400.00 USD annually (base salary)

Tools

C++, AWS, EC2, ECS, CloudWatch, S3, Lambda, PyTorch

AI focus

Build runtime libraries, drivers, and profiling tools for machine-learning applications and AI accelerators; improve ML kernel and framework performance and enable profiling across frameworks such as PyTorch, JAX, and XLA.

Model-written summary · the original posting is authoritative

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