Software Development Engineer II, EFA Network - Annapurna Labs

About the role

Want to help make the next generation of Machine Learning in the cloud possible? Do you have a laser focus on performance in your code? We want to talk to you!

We own the user-space software that makes the Elastic Fabric Adapter (EFA) network card work for Machine Learning (ML) and High-Performance Computing (HPC) customers on AWS. Across multiple projects written in C, our team enables customers to network thousands of GPU and CPU instance types to handle the toughest clustered workloads. Be a part of a dynamic, fast-paced group that has a big impact every day on the hottest companies doing AI and HPC today.

Key job responsibilities

• You will write the highest-performing code in C for multiple open source projects supporting EFA, such as Libfabric and Open MPI.
• You will work with multiple teams in the stack to invent new APIs for the latest concepts in networking in the cloud.
• Dive deep into how your customers are doing collectives and messaging at high bandwidth and low latency.
• Provide expert-level support to some of the biggest names in AI in the world.

A day in the life

• Start from the needs of your customer and invent new ways of cutting the occupancy of the software stack for EFA.
• Get your peers and stakeholders on board with excellent written designs.
• Write comprehensive tests to drive the development of new features and guard against regressions.
• Work with our ML Infrastructure team to see your products perform on 100s and 1000s of top-end machine clusters.

Basic Qualifications

• 3+ years of non-internship professional software development experience
• 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience
• 3+ years of professional experience programming high-performance software in C, ideally as part of an Open Source project

Preferred Qualifications

• 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
• Bachelor's degree in computer science or equivalent
• Experience developing in a network software stack, with a focus on cutting occupancy to the barest minimum number of instructions

Additional information

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If the country/region you're applying in isn't listed, please contact your Recruiting Partner.

Pay and benefits

The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave.

USA, WA, Seattle - 143,700.00 - 194,400.00 USD annually