Amazon
Sr Software Development Engineer, Neuron Collectives, Annapurna Labs
$193K–$262K
Cupertino, CA · Senior · Full-time · Posted May 28, 2026
- Experience
- 5+ years building complex software systems successfully delivered to customers; 5+ years contributing to system architecture and design.
- Education
- Bachelor's degree in computer science or equivalent; master's degree in computer science or equivalent preferred.
About the job
Optimizes C/C++ collective communication operations and training topologies for AWS Trainium, working across software, hardware, DMA, firmware, and data-center-scale AI systems.
What you'll do
- Enhance and optimize collective algorithms and topologies to improve training performance and bandwidth utilization.
- Analyze processor, DMA, firmware, and workload metrics to identify bottlenecks.
- Develop C/C++ implementations, analysis frameworks, and automation, and co-optimize software with Trainium silicon.
What they ask for
- Bachelor's degree in computer science or equivalent.
- 5+ years of experience building complex software systems successfully delivered to customers.
- 5+ years of experience contributing to architecture and design of new and existing systems, including design patterns, reliability, and scaling.
Nice to have
- Master's degree in computer science or equivalent.
- Familiarity with collective communication algorithms (e.g., all-reduce, all-gather) or distributed training frameworks.
Pay
193,300.00 - 261,500.00 USD annually (base salary)
Tools
C/C++, Neuron Explorer, DMA, firmware, Trainium
AI focus
Optimizes collective operations and topologies that scale AI model training across AWS Trainium chips and data centers.
Model-written summary · the original posting is authoritative