Senior Software Engineer, AI Training Collectives

Amazon; Cupertino (CA); On-site; USD 193,300 - 261,500; Full time

Benefits offered by this job

Health insurance

401(k) matching

Paid time off

Parental leave

Job summary

Amazon is looking for a Sr Software Development Engineer for Annapurna Labs in Cupertino, CA. You will enhance collective algorithms and optimize compute operations for AWS Trainium, working closely with hardware teams. This role requires extensive experience in software development and familiarity with collective communication algorithms.

The position offers a salary range from $193,300 to $261,500 annually, along with comprehensive benefits including health insurance, 401(k) matching, and paid time off.

Qualifications

• 5+ years of experience building complex software systems delivered to customers.
• Experience contributing to the architecture and design of new and current systems.
• Familiarity with distributed training frameworks.

Responsibilities

• Enhance collective algorithms and topologies for optimal training performance.
• Use tools like Neuron Explorer to identify bottlenecks.
• Optimize collective operations to scale AI compute across the data center.

Skills

C/C++

Collective communication algorithms

Software architecture and design

Education

Bachelor's degree in computer science or equivalent

Master's degree in computer science or equivalent

Job description

Amazon is looking for a Sr Software Development Engineer for Annapurna Labs in Cupertino, CA. You will enhance collective algorithms and optimize compute operations for AWS Trainium, working closely with hardware teams. This role requires extensive experience in software development and familiarity with collective communication algorithms.

The position offers a salary range from $193,300 to $261,500 annually, along with comprehensive benefits including health insurance, 401(k) matching, and paid time off.