Senior ML/HPC Networking & Systems Engineer

Amazon; Cupertino (CA); On-site; USD 150,000 - 210,000; Full time

Job summary

Amazon is seeking an experienced software engineer to work on distributed AI/ML systems within Annapurna Labs, contributing to EC2 infrastructure by building high-performance, scalable software in C/C++. You will collaborate with hardware and RTL engineers to optimize networking for ML/HPC workloads.

The role offers mentorship opportunities, cross-functional collaboration, and exposure to largest-scale AWS clusters and customers.

Qualifications

• 5+ years of non-internship professional software development experience.
• 5+ years of programming with C/C++.
• 5+ years of leading design or architecture of new and existing systems.
• 5+ years of full software development life cycle, including coding standards and reviews.
• Experience as a mentor, tech lead or leading an engineering team.

Responsibilities

• Develop distributed AI/ML systems for scaling across accelerators and servers.
• Contribute to Linux/kernel-level performance and reliability improvements.

Skills

C/C++

Linux

Distributed systems

Performance optimization

Mentorship

Education

Bachelor's degree in computer science or equivalent

Tools

NCCL

NVSHMEM

NIXL

Job description

Amazon is seeking an experienced software engineer to work on distributed AI/ML systems within Annapurna Labs, contributing to EC2 infrastructure by building high-performance, scalable software in C/C++. You will collaborate with hardware and RTL engineers to optimize networking for ML/HPC workloads.

The role offers mentorship opportunities, cross-functional collaboration, and exposure to largest-scale AWS clusters and customers.