Senior Software Development Engineer, Annapurna Labs, Trainium Collectives

Annapurna Labs, an integral part of AWS, develops hardware and software components for EC2 infrastructure, with every EC2 instance running their hardware designs.
$151,300 - $261,500
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Enterprise SaaS

Description For Senior Software Development Engineer, Annapurna Labs, Trainium Collectives

Amazon Web Services (AWS) is seeking a Senior Software Development Engineer to join Annapurna Labs, a crucial component of AWS that develops hardware and software for EC2 infrastructure. This role focuses on distributed AI/ML systems, specifically working on collective operations that enable AI scaling across multiple accelerators and servers.

The position requires expertise in C/C++ programming, Linux systems, and kernel development. The ideal candidate will have experience with embedded systems and high-speed networking or HPC interconnects. This role puts you at the forefront of AI/ML development, working on features for the largest clusters and AI models.

Working at Annapurna Labs means joining a diverse, international team where you'll collaborate with infrastructure experts, hardware engineers, RTL engineers, scientists, and architects. The organization values mentorship, work-life balance, and continuous learning. You'll have opportunities to work with principal-level engineers and directors, with clear paths for career growth.

The compensation package includes a competitive base salary ranging from $151,300 to $261,500 depending on location, plus equity, sign-on payments, and comprehensive benefits. The team emphasizes knowledge-sharing, mentorship, and thorough code reviews, making it an ideal environment for professional growth.

AWS values diverse experiences and maintains an inclusive culture through employee-led affinity groups and ongoing learning experiences. The company supports work-life harmony and provides extensive resources for career development. This role offers the unique opportunity to work on cutting-edge AI/ML technology while being part of Amazon's larger mission to be Earth's Best Employer.

Last updated 13 days ago

Responsibilities For Senior Software Development Engineer, Annapurna Labs, Trainium Collectives

  • Work on distributed AI/ML systems
  • Develop collective operations for AI scaling across multiple accelerators & servers
  • Write performant code in C/C++
  • Work with Linux kernels and embedded systems
  • Collaborate with infrastructure experts, hardware engineers, and RTL engineers
  • Mentor new and junior engineers

Requirements For Senior Software Development Engineer, Annapurna Labs, Trainium Collectives

Linux
  • 5+ years of non-internship professional software development experience
  • 5+ years of programming with at least one software programming language experience
  • 5+ years of leading design or architecture experience
  • 5+ years of full software development life cycle experience
  • Experience as a mentor, tech lead or leading an engineering team

Benefits For Senior Software Development Engineer, Annapurna Labs, Trainium Collectives

Medical Insurance
401k
  • Medical Insurance
  • 401k

Interested in this job?

Jobs Related To Amazon Senior Software Development Engineer, Annapurna Labs, Trainium Collectives

Software Development Engineer II, Expansion Team

Senior Software Engineer position at Amazon's Middle East and North Africa team, focusing on distributed systems and technical leadership.

Sr. Software Development Engineer - Connectivity, Amazon Devices

Senior Software Development Engineer position at Amazon Lab126, focusing on connectivity solutions for smart homes and enterprises, requiring 5+ years of experience in software development and system architecture.

Sr. Software Development Engineer - Connectivity, Amazon Devices

Senior Software Engineer role at Amazon Lab126 focusing on connectivity solutions for smart homes and enterprises, requiring 5+ years of experience in software development and system architecture.

Sr. Software Dev Engineer, Kuiper SDN Team

Senior Software Development Engineer position at Amazon's Project Kuiper, developing network control plane systems for a global satellite-based internet service.

Senior Software Development Engineer, Amazon MSK

Senior Software Development Engineer position for Amazon MSK team, building and maintaining large-scale Apache Kafka streaming services.