Amazon
Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training
$168K–$262K
Seattle, WA +1 · Senior · Full-time · Posted Aug 10, 2026
- Location
- Seattle, WA
Cupertino, CA - Experience
- 5+ years in each of the listed software development, programming, design or architecture, and full software development life cycle areas.
- Education
- Bachelor's degree in computer science or equivalent; master's degree in computer science or equivalent preferred.
About the job
Leads development and optimization of distributed model training on AWS Trainium, improving performance across the Neuron software stack and helping enable customer models.
What you'll do
- Optimize Trainium training throughput, model FLOPs utilization, and time to convergence across the Neuron stack.
- Enable and tune large-scale workloads, including model bring-up, parallelism strategies, profiling, and performance fixes across software layers.
- Mentor engineers, collaborate with customers and cross-functional teams, and contribute upstream and to future Trainium architecture requirements.
What they ask for
- Bachelor's degree in computer science or equivalent.
- 5+ years of non-internship professional software development experience, programming experience, system design or architecture experience, and full software development life cycle experience.
- Experience as a mentor, tech lead or leading an engineering team.
- Experience in machine learning, data mining, information retrieval, statistics or natural language processing.
Nice to have
- Master's degree in computer science or equivalent; experience in computer architecture.
- Software engineering expertise with Pytorch/Jax/Tensorflow, Distributed libraries and Frameworks, End-to-end Model Training.
Pay
USD 193,300.00 - 261,500.00 annually base salary; USD 168,100.00 - 227,400.00 annually base salary
Tools
AWS Neuron, Trainium, PyTorch, JAX, Neuron compiler, Neuron runtime, Tensorflow
AI focus
Enables and optimizes large-scale distributed machine learning and Generative AI model training on Trainium, including model bring-up, parallelism, numerics, and performance tuning.
Model-written summary · the original posting is authoritative