Software Development Engineer (AI)

TL;DR

Software Development Engineer (AI/ML): Building and optimizing distributed AI/ML systems, focusing on collective operations that scale across multiple accelerators and servers. Focus on low-level C/C++ development, Linux kernels, high-speed networking, and HPC interconnects for large-scale AI models.

Location: Must be based in the US, compensation reflects US geographic markets.

Salary: $129,300–$223,600/year

Company

Annapurna Labs, an integral part of AWS, develops hardware and software components critical for EC2 infrastructure and specializes in optimizing the AWS customer experience.

What you will do

• Work on distributed AI/ML systems and collective operations.
• Scale AI across multiple accelerators and servers.
• Develop features for the largest clusters and AI models.
• Collaborate with infrastructure experts, hardware engineers, RTL engineers, scientists, and architects.
• Iterate fast and deliver meaningful solutions at scale.

Requirements

• 3+ years of non-internship professional software development experience.
• 2+ years of non-internship design or architecture experience.
• Experience programming with at least one software programming language.
• Solid knowledge of Linux, kernels, and performant C/C++ code.
• Experience with high-speed networking or HPC interconnects is highly valued.

Nice to have

• 3+ years of full software development life cycle experience.
• Bachelor's degree in computer science or equivalent.
• Experience with embedded systems.

Culture & Benefits

• Flexible working hours and respect for work-life balance.
• Environment that celebrates knowledge-sharing and mentorship.
• Ongoing learning experiences and career growth opportunities.
• Comprehensive medical, financial, and other benefits.
• Inclusive team culture with affinity groups.