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.
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.
Builds AI. The job is building AI systems.
- ●●●● Builds AI80 to 100
- ●●●○ Works on AI60 to 79
- ●●○○ Uses AI40 to 59
- ●○○○ 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
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
- What's a project where you used Python hands-on?
- Walk me through how you've used C in your day-to-day work.
- What are the limits of AWS that you've run into, and how did you work around them?
- What's a project where you used Nccl hands-on?
- 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