JobsLead Engineer, ML Network Stack - Annapurna Labs
Amazon logo

Lead Engineer, ML Network Stack - Annapurna Labs

Amazon

Location

Cupertino, CA

Type

Full-time

Posted

5/5/2026

Compensation

$168,100 - $261,500 per year

Undergraduate with 5+ Years of Experience
Approval 98.6%·Filings 19,451·New hires 10,113·
👑 Elite Sponsor
·FY 2025

Job description

We are looking for an experienced engineer and technical leader to join our team at Annapurna Labs, which focuses on the network stack for EC2 distributed AI/ML systems. The role involves developing support for various frameworks and communication libraries while ensuring high performance and reliability. Candidates should have a solid understanding of Linux, networking, and coding practices, with a preference for those experienced in embedded systems and high-speed networking. This position offers the opportunity to grow into a technical manager role while working on cutting-edge AI/ML solutions.

Requirements

  • 5+ years of non-internship professional software development experience
  • 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 operations
  • 3+ years as a mentor, tech lead or leading engineering teams
  • 3+ years experience in SW/HW Co-Design
  • Bachelor's degree in computer science or equivalent
  • Experience creating automated dashboards and visualization such as Grafana

Responsibilities

  • Lead the engineering team that builds and maintains infrastructure for monitoring and reporting on functionality and performance of testing workloads.
  • Utilize internal Amazon CI/CD tools and public AWS products to automate software delivery to customers.
  • Write Python code to manage large clusters and run benchmarks for ML and HPC workloads.
  • Use AWS Managed Grafana and Athena to analyze performance data and create dashboards.
  • Invent mechanisms to alert developers to functional and performance regressions.
  • Manage complex infrastructure across various instance types and software stacks.

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.

Is this posting expired or inaccurate?