Level

Amazon

Software Development Engineer I – AI/ML Network Infrastructure, Annapurna Labs

Amazon is hiring a Software Development Engineer I – AI/ML Network Infrastructure, Annapurna Labs in Cupertino, United States. It pays $148k a year and Level rates it ; you can apply on Amazon.

Level's read on this job

  • Level rates this software engineer role Builds AI ●●●●: the job is building AI systems.
  • Tools and skills the posting cites include Python, C and AWS.
  • Level classifies it as a junior role.
  • Level lists 22,000+ other open roles at Amazon.
  • The role was posted in the last few days.

AI in this role

Develop high-performance C/C++ network infrastructure software for distributed AI/ML GPU clusters.

pythoncawsncclnvshmemnixlgrafanaathena
We're looking for a talented early-career engineer to join our team that owns the network stack for EC2 distributed AI/ML systems. You'll work on software that enables the world's largest AI models to train across massive GPU clusters, developing support for communication libraries and frameworks like NCCL, NVSHMEM, and NIXL.

This is a ground-floor opportunity to work at the intersection of high-performance computing, networking, and machine learning infrastructure - building the systems that power the largest AI workloads in the cloud.

Key job responsibilities
- Write high-performance C/C++ code for network communication libraries running on custom AWS hardware
- Build and maintain infrastructure that monitors functionality and performance of large-scale AI/ML workloads
- Develop automation using Python and AWS tools (CI/CD, Grafana, Athena) to test, benchmark, and deliver software to customers
- Design mechanisms to detect functional and performance regressions before they reach production
- Work across many instance types, software stacks, and Linux environments

Basic qualifications

Are 18 years of age or older
Experience with at least one modern language such as Java, Python, C++, or C# including object-oriented design
Experience with at least one general-purpose programming language such as Java, Python, C++, C#, Go, Rust, or TypeScript
Experience with data structure implementation, basic algorithm development, and/or object-oriented design principles
Experience using AI-assisted development tools in academic or professional settings
Bachelor's degree or above in Computer Science, Computer Engineering, or related fields
Strong proficiency in C/C++
Solid coursework or project experience in: 1/ Operating Systems (Linux internals, kernel concepts, memory management) 2/ Parallel Computer Architecture (multi-threading, SIMD, GPU programming, cache coherence) 3/ Distributed Systems (consensus, message passing, fault tolerance, scalability)
Familiarity with Linux development environments and toolchains

Preferred qualifications

Experience from a technical internship
Experience in optimization mathematics such as linear programming and nonlinear optimization
Experience with distributed, multi-tiered systems, algorithms, and relational databases
Work 40 hours/week, and overtime as required
Experience from previous technical internship(s) or demonstrated project experience
Experience with Cloud platforms (preferably AWS), database systems (SQL and NoSQL), AI tools for development productivity, contributing to open-source projects, and/or version control systems
Internship experience in ML communications, HPC networking, or RDMA/high-speed interconnects
Exposure to network programming (sockets, MPI, collective communication patterns)
Experience with performance profiling and optimization
Familiarity with GPU programming (CUDA) or hardware-software co-design
Contributions to open-source projects in systems, networking, or HPC

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: Job duties for this position include: work safely and cooperatively with other employees, supervisors, and staff; adhere to standards of excellence despite stressful conditions; communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service; and follow all federal, state, and local laws and Company policies. Criminal history may have a direct, adverse, and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above, as well as the abilities to adhere to company policies, exercise sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation. Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

The starting pay for this position is listed below. Starting Day 1 of employment, Amazon offers EAP, Mental Health Support, Medical Advice Line, 401(k) matching. Learn more about our benefits at https://hiring.amazon.com/why-amazon/benefits. 



Cupertino, CA, USA - 148,400.00 USD Annually

How we rate this

Software Development Engineer I – AI/ML Network Infrastructure, Annapurna Labs at Amazon rates 90 out of 100 for how much of the daily work is AI. That makes it Builds AI (AI Level 4 of 4). The level is about AI in the job, not seniority.

Classification

Builds AI. The job is building AI systems.

  1. ●●●● Builds AI80 to 100
  2. ●●●○ Works on AI60 to 79
  3. ●●○○ Uses AI40 to 59
  4. ●○○○ Little AI0 to 39

Levels come from how often the tools, models and workflows of the role are named in the posting itself. Read the description and count the AI tools, models and workflows it names.

Prepare for this job

Built from this posting and from similar jobs on Level.

Skills and AI tools this role asks for

PythonCAWSNcclNvshmemNixlGrafanaAthena

Also common in software engineer jobs at this level

Pay

This job starts at
$148k
Typical software engineer job
$168k

This posting starts 12% below the typical software engineer job. See software engineer jobs

Questions you could be asked

  1. What's a project where you used Python hands-on?
  2. Walk me through how you've used C in your day-to-day work.
  3. What are the limits of AWS that you've run into, and how did you work around them?
  4. What's a project where you used Nccl hands-on?
  5. Walk me through how you've used Nvshmem in your day-to-day work.

Adapt your resume

  • List these exact terms on your resume: Python, C, AWS, Nccl, and Nvshmem. An applicant tracking system matches the wording, not the idea.
  • Attach one line of real, concrete experience to at least one of them. A tool named with nothing behind it rarely survives a human read.
  • Lead with what you built, trained or shipped. This role is judged on the AI system itself, not the tools around it.

Want an expert to read your CV for this job?

Free. Send your CV and the role you want next. We reply by email within 2 to 4 business days.

Get new software engineer jobs (Builds AI ●●●●) by email

One email a week with the new software engineer jobs (Builds AI ●●●●), each rated for how much AI is in the work. No recruiter spam, unsubscribe in one click.

Free. One email a week. Unsubscribe in one click.

Similar roles

Software Engineering roles that build AI, at other companies.

What kind of AI work fits you?

Answer 12 practical questions in about three minutes. Get a simple profile, the work it points to, and live roles to explore next.

Find my next step

More jobs at Amazon

More software engineer jobs

Related searches

Same AI level

Jobs by city