Amazon
Sr. Software Engineer- AI/ML, AWS Neuron
$168K–$227K
Seattle, WA · Senior · Full-time · Posted Sep 7, 2026
- Experience
- 5+ years of non-internship professional software development experience; 5+ years leading design or architecture of new and existing systems.
- Education
- Bachelor's degree.
About the job
Build and optimize distributed inference capabilities, models, kernels, and system performance for AWS Neuron on Trainium and Inferentia accelerators.
What you'll do
- Lead development of distributed inference support for PyTorch and optimize models for AWS accelerators.
- Design, implement, profile, and optimize ML systems, kernels, and model operations across the development lifecycle.
- Build model onboarding infrastructure and collaborate with customers and cross-functional teams to improve performance.
What they ask for
- Bachelor's degree.
- 5+ years of non-internship professional software development experience.
- 5+ years of experience leading design or architecture of new and existing systems.
- Knowledge of Python and/or C++; experience debugging, profiling, and optimizing large-scale systems, including system performance, memory management, and parallel computing.
Nice to have
- Master's degree in computer science or equivalent.
- Knowledge of ML model architecture and inference; hands-on PyTorch, multi-node scaling, or CUDA/Triton kernel experience.
Pay
168,100.00 - 227,400.00 USD annually (base salary)
Tools
Python, C++, PyTorch, NCCL, CUDA, Triton, Trainium, Inferentia
AI focus
Develop distributed inference and optimize generative AI models, kernels, and ML workloads for AWS custom accelerators.
Model-written summary · the original posting is authoritative