JobsLead Software Engineer, ML Network Stack - Annapurna Labs
Amazon logo

Lead Software Engineer, ML Network Stack - Annapurna Labs

Amazon

Location

Cupertino, CA

Type

Full-time

Posted

7/31/2026

Compensation

$193,300 - $261,500 per year

Undergraduate with 5+ Years of Experience
H-1B FY202699.1% approval+9% YoY
👑 Elite sponsor

Job description

We are looking for an experienced engineer and technical leader to join our team focused on the network stack for EC2 distributed AI/ML systems. The team works with various frameworks and communication libraries, emphasizing Linux, networking, and efficient coding. Candidates should have a strong background in high-speed networking or HPC/RDMA interconnects. This role offers the opportunity to lead a diverse team and contribute to foundational technologies within AWS.

Requirements

  • 5+ years of leading design or architecture of new and existing systems experience
  • 5+ years of full software development life cycle experience, including coding standards and code reviews
  • Experience as a mentor or tech lead in the last 3 years
  • 5+ years of experience with programming languages C or C++
  • Bachelor's degree in computer science or equivalent
  • Experience working with ML Communication Libraries or RDMA Networking

Responsibilities

  • Lead the team working across the NVIDIA ML communication stack for enabling NVIDIA GPUs to work with AWS EC2 machines.
  • Direct work to ensure the team delivers functions and features required for the latest ML workloads.
  • Support new team members through knowledge-sharing and mentorship.
  • Participate in one-on-one mentoring and conduct thorough code reviews.
  • Help create an inclusive team culture that values diverse experiences.

Benefits

  • Employees at Amazon are often offered comprehensive health benefits—including multiple medical plan options (no pre-existing condition exclusions, 100% covered in-network preventive care), dental and vision plans, a 24/7 medical advice line from day one, expert second-opinion services, and broad mental-health support with several free counseling sessions (including pediatric). Financial wellness typically includes a 401(k) with company match (up to 2%), Restricted Stock Units (equity), FSAs, an emergency savings program, product and partner discounts, and even college-savings and home-purchase programs. Overall, the package is designed to support employees and their families’ health, finances, and day-to-day life.

H-1B filing history

Public USCIS petition and DOL LCA counts · latest USCIS FY2026, LCA FY2026

Filing entity: Amazon Com Services Llc

As of Aug 23, 2026

Initial approvals

3,290

FY2026

Approval rate

99.1%

FY2026

LCA certified

32,937

FY2026

Entry-level share

10.6%

FY2026

Initial approvals YoY

+9%

Trend

LCA certified YoY

-44%

Trend

Initial approvals by fiscal year

Approval rate by fiscal year

Continuing vs initial approvals

LCA certified positions by quarter

LCA certified positions by fiscal year

Based on public USCIS and DOL filings; not a sponsorship guarantee.

Is this posting expired or inaccurate?