Amazon
Software Engineer- AI/ML, Amazon Neuron Training
$144K–$224K
Seattle, WA +1 · Full-time · Posted Aug 13, 2026
- Location
- Seattle, WA
Cupertino, CA - Experience
- 3+ years of non-internship professional software development; 2+ years of non-internship design or architecture experience. Preferred: 3+ years of full software development life cycle experience.
- Education
- Bachelor's degree in computer science or equivalent (preferred).
About the job
Optimizes distributed training throughput on AWS Trainium by profiling workloads and improving performance across the Neuron software stack.
What you'll do
- Tune large-scale training workloads across PyTorch, Neuron, and Trainium, collaborating with compiler and runtime teams.
- Own parallelism strategies and apply reduced-precision formats to improve training performance.
- Profile end to end, drive fixes to the responsible software layer, and contribute performance requirements and changes to open-source frameworks.
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.
- Bachelor's degree in computer science or equivalent.
Pay
165,200.00 - 223,600.00 USD annually (base salary); 143,700.00 - 194,400.00 USD annually (base salary)
Tools
AWS Trainium, AWS Neuron, PyTorch, JAX
AI focus
Improves performance and efficiency of large-scale AI model training on AWS Trainium, including distributed training, parallelism, and reduced-precision formats.
Model-written summary · the original posting is authoritative