Lead Engineer, ML Network Stack - Annapurna Labs; AWS

Key Responsibilities

• Lead the engineering of infrastructure that monitors and reports on functionality and performance of massive testing workloads run at scale.
• Use internal Amazon CI/CD tools, Linux, and public AWS products to automate delivery of software to customers and save developer time.
• Write Python code to effortlessly spin up large clusters and run benchmarks and applications for ML and HPC workloads.
• Use AWS Managed Grafana and Athena to analyze performance data generated by workloads and create dashboards for developers and stakeholders.
• Invent automatic mechanisms to alert developers to functional and performance regressions before they reach customers.
• Manage the complexity of infrastructure covering many instance types, software stacks, Linux operating systems, cutting‑edge releases, and make it easy to evolve.

• 5+ years of non‑internship professional software development experience.
• 5+ years leading design or architecture (design patterns, reliability and scaling) of new and existing systems.
• 5+ years of full software development life cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
• 3+ years as a mentor, tech lead, or leading engineering teams.
• 3+ years of SW/HW co‑design experience.

• Bachelor’s degree in computer science or equivalent.
• Experience creating automated dashboards and visualizations (e.g., Grafana).

Amazon offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and optional Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at (Use the "Apply for this Job" box below)..

USA, CA, Cupertino – $ – $ USD annually.

USA, WA, Seattle – $ – $ USD annually.

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:
Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.