EFA Network Software Engineer I, Annapurna Labs

Amazon

Seattle (WA)

On-site

USD 110,500 - 160,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Job summary

Amazon is seeking engineers to write high-performance C code for the Elastic Fabric Adapter (EFA) user-space stack, enabling hosts to network thousands of GPU/CPU instances for ML and HPC workloads.

You will contribute to Libfabric/Open MPI and collaborate across teams to design APIs, optimize occupancy, and support major AI customers. The role emphasizes hands-on performance work and scalable networking in the cloud.

Qualifications

  • Bachelor's degree or above in computer science, computer engineering, or related field, or Bachelor's degree
  • Strong programming skills in C (coursework, projects, or internship experience)
  • Familiarity with systems programming concepts (memory management, performance optimization, Linux)
  • Experience with source control (Git), code reviews, and build processes (internship experience counts)

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.

Skills

C programming
Linux
Git
Performance optimization
Systems programming

Education

Bachelor's degree in CS/CE

Tools

Git
Build systems
Libfabric/Open MPI familiarity

Job description

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.

About the team

We are a fast-paced team that owns the user-space software stack for EFA. As part of Annapurna Labs in AWS we are very nimble, paying careful attention to what the AI industry is going to try next, and having our products ready. We focus heavily on automation, confining operations to the most interesting problems as customers continuously experiment with what our network can do. Our team is a place of growth, concentrating on your career and goals and motivating you to achieve your highest potential.

Basic Qualifications
  • Bachelor's degree or above in computer science, computer engineering, or related field, or Bachelor's degree
  • Strong programming skills in C (coursework, projects, or internship experience)
  • Familiarity with systems programming concepts (memory management, performance optimization, Linux)
  • Experience with source control (Git), code reviews, and build processes (internship experience counts)
Preferred Qualifications
  • Contributions to open source projects (Libfabric, Open MPI, or similar)
  • Coursework or experience in computer networking (protocols, latency optimization, distributed systems)
  • Experience writing performance-critical code with a focus on minimizing instruction count and occupancy
  • Familiarity with HPC or ML infrastructure concepts
  • Experience with the full software development lifecycle including testing and operations

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

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. Learn more about our benefits at https://amazon.jobs/en/benefits .

USA, WA, Seattle - 110,500.00 - 160,000.00 USD annually

Important FAQs for current Government employees

Before proceeding, please review the following FAQs

https://www.amazon.jobs/en/faqs#faqs-for-us-government-employees

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

EFA Network Software Engineer I, Annapurna Labs
EFA Network Software Engineer I, Annapurna Labs

Amazon Web Services (AWS) • Seattle (WA)

On-site
USD 110,000 - 160,000
Sr. Manager, Software Development, ML Network Stack - Annapurna Labs
Sr. Manager, Software Development, ML Network Stack - Annapurna Labs

Amazon • Seattle (WA)

On-site
USD 220,000 - 298,000
Health insurance
401(k) matching
Paid time off
+1
Sr. Manager, Software Development, ML Network Stack - Annapurna Labs
Sr. Manager, Software Development, ML Network Stack - Annapurna Labs

Socket.dev • Seattle (WA)

On-site
USD 220,000 - 298,000
Health insurance
401(k) matching
Paid time off
+1
Lead Software Engineer, ML Network Stack - Annapurna Labs
Lead Software Engineer, ML Network Stack - Annapurna Labs

Amazon • Cupertino (CA)

On-site
USD 193,000 - 262,000
Sr. Software Development Engineer, HPC/ML Networking Engineer, Annapurna Labs
Sr. Software Development Engineer, HPC/ML Networking Engineer, Annapurna Labs

Socket.dev • Cupertino (CA)

On-site
USD 193,300 - 261,500
EFA Networking Engineer for ML & HPC (C)
EFA Networking Engineer for ML & HPC (C)

Amazon Web Services (AWS) • Seattle (WA)

On-site
USD 110,000 - 160,000
Software Development Engineer, Annapurna Labs, Elastic Collectives
Software Development Engineer, Annapurna Labs, Elastic Collectives

Amazon Web Services (AWS) • Cupertino (CA)

On-site
USD 165,000 - 224,000
Sr. Software Development Engineer, Annapurna Labs
Sr. Software Development Engineer, Annapurna Labs

Amazon • Cupertino (CA)

On-site
USD 193,000 - 262,000
Health insurance
401(k) matching
Paid time off
+1
Manager, Software Development, ENA Express Team, Annapurna Labs
Manager, Software Development, ENA Express Team, Annapurna Labs

Amazon • Cupertino (CA)

On-site
USD 213,000 - 288,000
Lead Software Engineer, ML Network Stack - Annapurna Labs
Lead Software Engineer, ML Network Stack - Annapurna Labs

Amazon Web Services (AWS) • Cupertino (CA)

On-site
USD 193,000 - 262,000
RSUs
Comprehensive benefits
Health insurance