← All roles

Software Development Engineer I – AI/ML Network Infrastructure

Amazon· Annapurna Labs (U.S.) Inc.
Apply on company site
Cupertino, CA, USA$148.4kFull-timePosted 2w ago

About the role

Early-career software engineer role focusing on the network stack for EC2 distributed AI/ML systems, developing high-performance communication libraries and infrastructure to support training across large GPU clusters.

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

Qualifications

  • Bachelor's degree or above in Computer Science, Computer Engineering, or related fields
  • 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

Skills

Required:C/C++Operating SystemsParallel Computer ArchitectureDistributed SystemsLinux development environments
Nice to have:Python

Benefits

  • EAP, Mental Health Support, Medical Advice Line, 401(k) matching
Last updated Oct 25, 2026. Interested? Apply on the company site.

About Amazon

Amazon.com, Inc. or its affiliates

liquid-ai-recruiting-target
Company profileWebsite

More open roles at Amazon