Software Development Engineer I – AI/ML Network Infrastructure, Annapurna Labs

Key job responsibilities

• Write high-performance C/C++ code for network communication libraries running on custom AWS hardware
• Build and maintain infrastructure that monitors functionality and performance of large-scale AI/ML workloads
• Develop automation using Python and AWS tools (CI/CD, Grafana, Athena) to test, benchmark, and deliver software to customers
• Design mechanisms to detect functional and performance regressions before they reach production
• Work across many instance types, software stacks, and Linux environments

Basic Qualifications

• Bachelor's or Master's degree in Computer Science, Computer Engineering, or related field (recent graduates welcome)
• Strong proficiency in C/C++
• Solid coursework or project experience in:
• Operating Systems (Linux internals, kernel concepts, memory management)
• Parallel Computer Architecture (multi-threading, SIMD, GPU programming, cache coherence)
• Distributed Systems (consensus, message passing, fault tolerance, scalability)

Preferred Qualifications

• Internship experience in ML communications, HPC networking, or RDMA/high-speed interconnects
• Familiarity with network programming (sockets, MPI, collective communication patterns)
• Experience with performance profiling and optimization
• Familiarity with GPU programming (CUDA) or hardware-software co-design

Company

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

Los Angeles County Applicants Only

• Job duties for this position include:
• work safely and cooperatively with other employees, supervisors, and staff;
• adhere to standards of excellence despite stressful conditions;
• communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service;
• follow all federal, state, and local laws and Company policies.

Criminal History

• Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Inclusive Culture

• Our inclusive culture empowers Amazonians to deliver the best results for our customers.

Final Compensation

• 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.

Benefits

• Comprehensive benefits including health insurance, 401(k) matching, paid time off, and parental leave.
• Learn more about our benefits at https://amazon.jobs/en/benefits.