[PHD] SDE - Systems, Runtime, and ML Infrastructure (AWS Custom Silicon), Annapurna Labs

Amazon

On-site 🏢 15 April
R&D, Innovation
Austin, United States 🇺🇸 Seattle, United States 🇺🇸 Cupertino, United States 🇺🇸

At AWS, we're pioneering the future of cloud computing and AI acceleration through innovative hardware-software co-design. Our teams within Annapurna Labs and AWS AI are creating the foundation for next-generation cloud infrastructure that powers thousands of customers worldwide, from cutting-edge startups to global enterprises.We operate at an unprecedented scale, designing custom silicon chips, advanced networking solutions, and ML accelerators that were unimaginable just a few years ago. Our work spans from the lowest levels of hardware abstraction to high-performance distributed training systems, creating unique opportunities for early-career engineers to make significant impact across multiple domains.Key job responsibilities- Develop and optimize software for custom hardware and ML infrastructure- Collaborate with hardware teams to understand and leverage chip architecture- Implement and improve networking, runtime, and system-level software- Assist in building and maintaining tools for profiling, monitoring, and debugging ML workloads- Contribute to the development of open-source ML frameworks and infrastructure projects- Participate in code reviews and implement best practices for software development- Learn and apply new technologies to solve complex engineering challengesAbout the teamCandidates will be routed to specific teams based on their interests and our current needs during the application process:- The Elastic Network Adapter (ENA) team revolutionizes EC2 core networking, enabling enhanced networking capabilities across AWS's most critical compute instances. Here, you'll work with networking protocols and high-performance drivers that power millions of cloud workloads.- Our AWS Neuron SDK team develops the complete software stack for custom ML accelerators (Inferentia and Trainium), democratizing access to AI infrastructure. This team bridges the gap between popular ML frameworks and custom hardware.- The Machine Learning Server Software team maintains and optimizes the world's most advanced ML servers, focusing on system-level software that ensures peak performance of AI workloads. While we don't work directly on ML algorithms, we build the critical infrastructure that makes ML possible at scale.- The SoC Hardware Abstraction Layer (HAL) team works at the intersection of hardware and software, developing the crucial middleware that manages our custom silicon chips. This team ensures our innovative hardware designs translate into reliable, high-performance solutions.

Apply now

Similar jobs

Mechanical Process Engineer Intern, 2025 Internship, Shanghai, China

On-site 🏢 05 December
R&D, Innovation
Shanghai, China 🇨🇳 Shenzhen, China 🇨🇳

GPU Firmware Engineer (RDSS Intern)

On-site 🏢 09 October
R&D, Innovation
Taipei, Taiwan 🇹🇼

芯片验证实习生 DV Intern (Jul - Dec 2025)

Remote 🌎🌍🌏 16 April
R&D, Innovation
Shanghai, China 🇨🇳

2025 Applied Science Internship - Computer Vision - United States, PhD Student Science Recruiting

On-site 🏢 02 October
R&D, Innovation
Seattle, United States 🇺🇸

Research Intern - High performance and Scalable Database Systems

Hybrid 🏡🏢 10 March
R&D, Innovation
California, United States 🇺🇸 Washington, United States 🇺🇸 Texas, United States 🇺🇸

Research Intern - Applied Speech Research

Remote 🌎🌍🌏 28 March
R&D, Innovation
California, United States 🇺🇸

Applied Scientist Intern, International Technology, 2025 Beijing

On-site 🏢 09 December
R&D, Innovation
Beijing, China 🇨🇳

Research Assistant - PFE 2025

On-site 🏢 19 March
R&D, Innovation
Casablanca, Morocco 🇲🇦

Student / Intern in Oracle Labs

On-site 🏢 22 March
R&D, Innovation
Zurich, Switzerland 🇨🇭

Applied Scientist Intern, Global CoreJP, 2025 Beijing

On-site 🏢 09 December
R&D, Innovation
Beijing, China 🇨🇳

Research Intern - Office of the Chief Scientific Officer

Hybrid 🏡🏢 11 December
R&D, Innovation
Washington, United States 🇺🇸 California, United States 🇺🇸

2025 Applied Science Internship - Gen AI & Large Language Models - United States, PhD Student Science Recruiting

On-site 🏢 02 October
R&D, Innovation
Seattle, United States 🇺🇸

GPU AI Kernel开发实习生 (Jul - Dec 2025)

On-site 🏢 New 🔥🔥
R&D, Innovation
Beijing, China 🇨🇳

[PhD] ML Compiler Engineer - Annapurna Labs, Annapurna Labs

On-site 🏢 15 April
R&D, Innovation
Seattle, United States 🇺🇸 Cupertino, United States 🇺🇸

Firmware Intern (Jul - Dec 2025)

On-site 🏢 11 April
R&D, Innovation
T´ai-pei, Taiwan 🇹🇼

2025 Applied Science Internship - United States - PhD Student Science Recruiting

On-site 🏢 15 October
R&D, Innovation
Seattle, United States 🇺🇸

Research Intern - MSR Inclusive Futures Team

Hybrid 🏡🏢 25 October
R&D, Innovation
Washington, United States 🇺🇸

Perf Engineering Intern - 2025

On-site 🏢 14 January
R&D, Innovation
Shanghai, China 🇨🇳

2025 ASIC Physical Design Engineer Intern, Annapurna Labs

On-site 🏢 03 October
R&D, Innovation
Austin, United States 🇺🇸 Cupertino, United States 🇺🇸

Research Intern - AI-Assisted Game Creation

Hybrid 🏡🏢 21 November
R&D, Innovation
Washington, United States 🇺🇸 California, United States 🇺🇸

Research Intern

On-site 🏢 02 April
R&D, Innovation
Canada 🇨🇦

Research Scientist, Generalist Embodied Agent Research - New College Grad 2025

Remote 🌎🌍🌏 12 November
R&D, Innovation
California, United States 🇺🇸 Remote, United States 🇺🇸

シニアリサーチャー（自然言語処理）｜Senior Researcher, Natural Language Processing

On-site 🏢 06 April
R&D, Innovation
Tokyo, Japan 🇯🇵

Research Intern - Applied Sciences Group (Computer Vision)

Hybrid 🏡🏢 06 January
R&D, Innovation
Washington, United States 🇺🇸

TiMi Singapore or Montreal- Level design intern (ancient China history research)

On-site 🏢 20 March
R&D, Innovation
Singapore 🇸🇬

PhD Research Intern, Generalist Embodied Agents Research - Summer 2025

Remote 🌎🌍🌏 14 April
R&D, Innovation
California, United States 🇺🇸 Remote, United States 🇺🇸

2025 Applied Scientist Internship, Amazon University Talent Acquisition

On-site 🏢 03 October
R&D, Innovation
London, United Kingdom 🇬🇧 Cambridge, United Kingdom 🇬🇧 Edinburgh, United Kingdom 🇬🇧

2025 Applied Scientist Internship, Amazon University Talent Acquisition

On-site 🏢 01 October
R&D, Innovation
Aachen, Germany 🇩🇪 Berlin, Germany 🇩🇪 Tuebingen, Germany 🇩🇪 Munich, Germany 🇩🇪

PhD Research Intern, Generalist Embodied Agents Research - Fall 2025

Remote 🌎🌍🌏 14 April
R&D, Innovation
California, United States 🇺🇸 Remote, United States 🇺🇸

Silicon Validation Engineer (RDSS Intern)

On-site 🏢 22 October
R&D, Innovation
Taipei, Taiwan 🇹🇼

Mixed Signal Design Engineer (RDSS Intern)

On-site 🏢 31 October
R&D, Innovation
Hsinchu, Taiwan 🇹🇼 Taipei, Taiwan 🇹🇼

2025 Applied Science Internship - United States, Undergrad Student Science Recruiting, Frontier AI & Robotics

On-site 🏢 04 February
R&D, Innovation
San Francisco, United States 🇺🇸

Research Sciences Resident (AI for Good) Intern

Hybrid 🏡🏢 04 March
R&D, Innovation
Nairobi City, Kenya 🇰🇪

Mixed Signal Analog Circuit Designer (RDSS Intern)

On-site 🏢 12 November
R&D, Innovation
Hsinchu, Taiwan 🇹🇼 Taipei, Taiwan 🇹🇼

ASIC Design Verification Engineer, Annapurna Labs

On-site 🏢 17 March
R&D, Innovation
Austin, United States 🇺🇸 Cupertino, United States 🇺🇸

Research Engineer (L4) - Member Lifecycle and Monetization

Remote 🌎🌍🌏 20 March
R&D, Innovation
USA - Remote

Mixed Signal Design Engineer (RDSS Intern)

On-site 🏢 15 November
R&D, Innovation
Taipei, Taiwan 🇹🇼 Hsinchu, Taiwan 🇹🇼

Research Sciences Intern (VISION)

Hybrid 🏡🏢 24 January
R&D, Innovation
Berlin, Germany 🇩🇪

Research Intern - Bio LLMs

Hybrid 🏡🏢 31 January
R&D, Innovation
Washington, United States 🇺🇸

System Design Power Validation Engineer - New College Grad

On-site 🏢 New 🔥🔥
R&D, Innovation
Taipei, Taiwan 🇹🇼

Silicon Design Intern (Jul - Dec 2025)

Remote 🌎🌍🌏 02 April
R&D, Innovation
Taipei City 115, Taiwan 🇹🇼

Product Development Engineer (RDSS Intern)

On-site 🏢 20 November
R&D, Innovation
Hsinchu, Taiwan 🇹🇼

2025 Co-Op: R&D Packaging

On-site 🏢 29 August
R&D, Innovation
Texas, United States 🇺🇸 New York, United States 🇺🇸 Illinois, United States 🇺🇸

Signal and Power Integrity Engineer (RDSS Intern)

On-site 🏢 06 November
R&D, Innovation
Taipei, Taiwan 🇹🇼 Hsinchu, Taiwan 🇹🇼

Mixed-Signal Circuit Design Engineer - New College Graduate

On-site 🏢 06 December
R&D, Innovation
Hsinchu, Taiwan 🇹🇼 Taipei, Taiwan 🇹🇼

2025 Applied Science Internship - United States - Master's Student Science Recruiting

On-site 🏢 07 October
R&D, Innovation
Seattle, United States 🇺🇸

游戏AI研究员

On-site 🏢 06 April
R&D, Innovation
Tokyo, Japan 🇯🇵

Server Platform Validation Intern

On-site 🏢 14 February
R&D, Innovation
Pulau Pinang, Malaysia 🇲🇾

Research Assistant - PFE 2025

On-site 🏢 19 March
R&D, Innovation
Casablanca, Morocco 🇲🇦

Student / Intern_Oracle Labs – Distributed Graph / ML in DB / Vector Database Systems

On-site 🏢 28 March
R&D, Innovation
Zurich, Switzerland 🇨🇭

2025 Research Science Internship - Robotics - United States, PhD Student Science Recruiting

On-site 🏢 15 November
R&D, Innovation
San Francisco, United States 🇺🇸

Service Process Innovation & Systems Support (INTERNSHIP)

Hybrid 🏡🏢 08 August
R&D, Innovation
Milano, Italy 🇮🇹

Design Verification Intern

On-site 🏢 12 February
R&D, Innovation
Pulau Pinang, Malaysia 🇲🇾

2025 Applied Scientist Internship, Amazon University Talent Acquisition

On-site 🏢 08 October
R&D, Innovation
Luxembourg 🇱🇺

Stagiaire en recherche- équité, responsabilité, transparence et éthique | Research Intern - Fairness, Accountability, Transparency, and Ethics (FATE)

Hybrid 🏡🏢 05 December
R&D, Innovation
Québec, Canada 🇨🇦

Speech Synthesis Intern

On-site 🏢 26 February
R&D, Innovation
Singapore 🇸🇬

2025 Applied Science Internship - Canada, PhD Student Science Recruiting

On-site 🏢 02 October
R&D, Innovation
Vancouver, Canada 🇨🇦 Pitt Meadows, Canada 🇨🇦 Toronto, Canada 🇨🇦 Burnaby, Canada 🇨🇦

Applied Scientist (北京 2024-25校招 International Technology Team)

On-site 🏢 14 August
R&D, Innovation
Beijing, China 🇨🇳

Student / Intern in Oracle Labs

On-site 🏢 05 March
R&D, Innovation
Zurich, Switzerland 🇨🇭

Hardware Development Engineer, Annapurna Labs

On-site 🏢 16 December
R&D, Innovation
Austin, United States 🇺🇸 Cupertino, United States 🇺🇸

2025 Applied Science Internship - United States, PhD Student Science Recruiting, Frontier AI & Robotics

On-site 🏢 11 December
R&D, Innovation
Seattle, United States 🇺🇸 San Francisco, United States 🇺🇸

Mixed Signal Design Engineer - New College Grad 2025

On-site 🏢 03 March
R&D, Innovation
California, United States 🇺🇸

Research Intern - Microsoft Teams CMD Labs

Hybrid 🏡🏢 03 April
R&D, Innovation
Washington, United States 🇺🇸

GPU Power Architect - New College Grad 2025

On-site 🏢 05 March
R&D, Innovation
California, United States 🇺🇸

2025 Operations Research Science Internship - United States, PhD Student Science Recruiting

On-site 🏢 02 October
R&D, Innovation
Seattle, United States 🇺🇸

Student / Intern in Oracle Labs App Dev Projects

On-site 🏢 27 November
R&D, Innovation
Zurich, Switzerland 🇨🇭

Cambridge Research Intern - MSR Intelligent Networked Systems

Hybrid 🏡🏢 30 January
R&D, Innovation
Cambridgeshire, United Kingdom 🇬🇧

Applied Research Intern - 2025

On-site 🏢 14 February
R&D, Innovation
Yerevan, Armenia

2025 Mechanical/Electrical Engineering Internship

On-site 🏢 01 November
R&D, Innovation
Milan, Italy 🇮🇹

Student / Intern_Oracle Labs

On-site 🏢 27 November
R&D, Innovation
Zurich, Switzerland 🇨🇭

Student / Intern in Oracle Labs

On-site 🏢 22 March
R&D, Innovation
Zurich, Switzerland 🇨🇭

Research Intern - Quantum Computing

Hybrid 🏡🏢 26 February
R&D, Innovation
Washington, United States 🇺🇸

Speech Synthesis Intern

On-site 🏢 02 March
R&D, Innovation
Singapore 🇸🇬

Speech Synthesis Intern

On-site 🏢 12 February
R&D, Innovation
London, United Kingdom 🇬🇧

Amazon Science Fellow, JP Manga

On-site 🏢 15 April
R&D, Innovation
Tokyo, Japan 🇯🇵

2025 Applied Scientist internship- UAE National, Amazon University Talent Acquisition

On-site 🏢 27 January
R&D, Innovation
Dubai, United Arab Emirates 🇦🇪

SRAM Circuit Design Engineer - New College Grad 2025

On-site 🏢 08 April
R&D, Innovation
California, United States 🇺🇸

Research Intern - Biomedical AI for Precision Health

Hybrid 🏡🏢 04 November
R&D, Innovation
Washington, United States 🇺🇸 Massachusetts, United States 🇺🇸 California, United States 🇺🇸

Vehicle Adaptation Intern - 2025

On-site 🏢 13 April
R&D, Innovation
Stuttgart, Germany 🇩🇪

Quantum Researcher PhD (Intern) United States

Remote 🌎🌍🌏 17 April
R&D, Innovation
California, United States 🇺🇸

Research Assistant - PFE 2025

On-site 🏢 19 March
R&D, Innovation
Casablanca, Morocco 🇲🇦

Intern Researcher in Computer Vision and Machine Learning

Hybrid 🏡🏢 05 March
R&D, Innovation
Cambridge, United Kingdom 🇬🇧

Research Intern - Mathematics

Hybrid 🏡🏢 28 October
R&D, Innovation
Massachusetts, United States 🇺🇸

2025 Applied Scientist Internship, Amazon University Talent Acquisition, Amazon University Talent Acquisition

On-site 🏢 20 February
R&D, Innovation
Paris, France 🇫🇷

Research and Development Intern

On-site 🏢 09 April
R&D, Innovation
Singapore 🇸🇬

2025 Applied Science Internship - Automated Reasoning - United States, PhD Student Science Recruiting

On-site 🏢 02 October
R&D, Innovation
Seattle, United States 🇺🇸

Research Scientist, Design Automation - New College Grad 2025

On-site 🏢 28 October
R&D, Innovation
Texas, United States 🇺🇸 California, United States 🇺🇸

Research Assistant - PFE 2025

On-site 🏢 19 March
R&D, Innovation
Casablanca, Morocco 🇲🇦

Design Verification Engineer (RDSS Intern)

On-site 🏢 06 November
R&D, Innovation
Hsinchu, Taiwan 🇹🇼

Research Intern - Formal Language Processing

Hybrid 🏡🏢 13 February
R&D, Innovation
Washington, United States 🇺🇸

2025 Quantum Research Science Internship - United States, PhD Student Science Recruiting

On-site 🏢 02 October
R&D, Innovation
Seattle, United States 🇺🇸

Research Intern - Multimodal AI Research

Hybrid 🏡🏢 27 January
R&D, Innovation
Washington, United States 🇺🇸 Massachusetts, United States 🇺🇸

Research Intern - Gray Systems Lab

Hybrid 🏡🏢 02 October
R&D, Innovation
Washington, United States 🇺🇸 Wisconsin, United States 🇺🇸 California, United States 🇺🇸

Research Intern - Applied Sciences Group (Computer Vision)

Hybrid 🏡🏢 04 March
R&D, Innovation
Washington, United States 🇺🇸

Research Intern - AI HW/SW Co-design

Hybrid 🏡🏢 10 January
R&D, Innovation
Washington, United States 🇺🇸 California, United States 🇺🇸

2025 Applied Scientist Internship, Amazon University Talent Acquisition

On-site 🏢 08 October
R&D, Innovation
Barcelona, Spain 🇪🇸 Madrid, Spain 🇪🇸

Student / Intern in Oracle Labs App Dev Projects

On-site 🏢 New 🔥🔥
R&D, Innovation
Zurich, Switzerland 🇨🇭

Share this job, spread the word!

Apply now

Similar jobs

Mechanical Process Engineer Intern, 2025 Internship, Shanghai, China

On-site 🏢 05 December
R&D, Innovation
Shanghai, China 🇨🇳 Shenzhen, China 🇨🇳

GPU Firmware Engineer (RDSS Intern)

On-site 🏢 09 October
R&D, Innovation
Taipei, Taiwan 🇹🇼

芯片验证实习生 DV Intern (Jul - Dec 2025)

Remote 🌎🌍🌏 16 April
R&D, Innovation
Shanghai, China 🇨🇳