Software Development Engineer I – AI/ML… at Amazon · Cupertino

Summary: Annapurna Labs, a subsidiary of Amazon, is seeking an early-career Software Development Engineer to join their team focused on AI/ML network infrastructure. The role involves developing high-performance software for communication libraries and maintaining infrastructure for large-scale AI workloads in the cloud. 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 Required 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: 1/ Operating Systems (Linux internals, kernel concepts, memory management) 2/ Parallel Computer Architecture (multi-threading, SIMD, GPU programming, cache coherence) 3/ Distributed Systems (consensus, message passing, fault tolerance, scalability) - Familiarity with Linux development environments and toolchains Preferred Qualifications: - Internship experience in ML communications, HPC networking, or RDMA/high-speed interconnects - Exposure to network programming (sockets, MPI, collective communication patterns) - Experience with performance profiling and optimization - Familiarity with GPU programming (CUDA) or hardware-software co-design - Contributions to open-source projects in systems, networking, or HPC Required Skills: C++, C, Operating Systems - Linux internals, Operating Systems - kernel concepts, Operating Systems - memory management, Parallel Computer Architecture - multi-threading, Parallel Computer Architecture - SIMD, Parallel Computer Architecture - GPU programming, Parallel Computer Architecture - cache coherence, Distributed Systems - consensus, Distributed Systems - message passing, Distributed Systems - fault tolerance, Distributed Systems - scalability, Linux development environments, Python, AWS CI/CD tools, Grafana, Athena, Network programming - sockets, Network programming - MPI, Network programming - collective communication patterns, Performance profiling, Performance optimization, CUDA, Hardware-software co-design Benefits: Sign-on payments, Restricted stock units (RSUs), 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, Parental leave