SDE I: AI/ML Network Infrastructure

Amazon; Cupertino (CA); On-site; USD 134,000 - 163,000; Full time

Job summary

Amazon is seeking an early-career engineer to join the team responsible for the EC2 network stack supporting distributed AI/ML systems. You will develop high-performance C/C++ code for network libraries on AWS hardware and build infrastructure to monitor large-scale AI workloads.

The role emphasizes collaboration across Linux environments, optimization, and exposure to GPU programming (CUDA) and MPI. It offers the chance to contribute to the world’s largest AI workloads in the cloud and grow

Qualifications

• Bachelor's degree in Computer Science, Computer Engineering, or related field.
• Strong proficiency in C/C++ and Linux development environments.
• Knowledge of operating systems, parallel computer architecture, and distributed systems.

Responsibilities

• Write high-performance C/C++ code for network communication libraries.
• Build and maintain infrastructure monitoring large-scale AI/ML workloads.
• Develop automation with Python and AWS tools for testing, benchmarking, and delivery.
• Design mechanisms to detect functional and performance regressions before production.
• Work across many instance types, software stacks, and Linux environments.

Skills

C/C++ proficiency

Linux internals

Parallel computing

Distributed systems

Education

Bachelor's degree in CS/CE or related fields

Tools

MPI

CUDA

Linux toolchains

Job description

Amazon is seeking an early-career engineer to join the team responsible for the EC2 network stack supporting distributed AI/ML systems. You will develop high-performance C/C++ code for network libraries on AWS hardware and build infrastructure to monitor large-scale AI workloads.

The role emphasizes collaboration across Linux environments, optimization, and exposure to GPU programming (CUDA) and MPI. It offers the chance to contribute to the world’s largest AI workloads in the cloud and grow