Software Development Engineer, Neuron Collectives, Annapurna Labs

Amazon

Cupertino, CA, US
Not specified; not specified; not specified
On-site
C/c++ programming expertise
Collective algorithm optimization
Dma and firmware interfacing
The Software Development Engineer role at Annapurna Labs, part of Amazon, focuses on optimizing collective operations for AWS Trainium, a critical initiative for AI training. The position requires expertise in C/C++ and involves collaboration with hardware teams to enhance performance across data centers

Job Summary

  • This role focuses on optimizing collective operations for AWS Trainium to power frontier AI models at Amazon scale.
  • Candidates will work closely with hardware teams to push maximum performance using C/C++ and analyze processor metrics.
  • The position offers opportunities to impact AI training infrastructure while growing technical breadth in a culture of inclusion.

Matching Summary

Match Score: 85

The Software Development Engineer role at Annapurna Labs, part of Amazon, focuses on optimizing collective operations for AWS Trainium, a critical initiative for AI training. The position requires expertise in C/C++ and involves collaboration with hardware teams to enhance performance across data centers.

Salary

Not specified; Not specified; Not specified

Skills & Requirements

Must-have

  • C/C++ programming expertise
  • Collective algorithm optimization
  • DMA and firmware interfacing
  • Neuron Explorer tool usage
  • Distributed system topologies

Nice-to-have

  • Hardware-software co-optimization experience
  • LLM training topology knowledge
  • Inclusive team culture participation
  • Mentorship and career growth focus

Key Requirements

  • Proficiency in C/C++ implementation
  • Experience with DMA and firmware interfaces
  • Knowledge of distributed computing topologies

Work Rights

Not specified

Tailored Resume

Cover Letter