[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

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

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

Research Intern - MSR Inclusive Futures Team

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

Research Intern - AI-Assisted Game Creation

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

Research Assistant - PFE 2025

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

Research Intern - Office of the Chief Scientific Officer

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

GPU Power Architect - New College Grad 2025

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

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

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

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

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

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 🇨🇦

Silicon Design Intern (Jul - Dec 2025)

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

Research Intern - Quantum Computing

Hybrid 🏡🏢 26 February
R&D, Innovation
Washington, 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 🇺🇸

Student / Intern_Oracle Labs

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

Applied Scientist Intern, Global CoreJP, 2025 Beijing

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

SRAM Circuit Design Engineer - New College Grad 2025

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

Student / Intern in Oracle Labs

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

Cambridge Research Intern - MSR Intelligent Networked Systems

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

Research Assistant - PFE 2025

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

Research Intern - AI HW/SW Co-design

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

PhD Research Intern, Generalist Embodied Agents Research - Summer 2025

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

Vehicle Adaptation Intern - 2025

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

Physical Design Backend Student

On-site 🏢 New 🔥🔥
R&D, Innovation
Tel Aviv, Israel 🇮🇱 Yokneam, Israel 🇮🇱

Perf Engineering Intern - 2025

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

Mixed-Signal Circuit Design Engineer - New College Graduate

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

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

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

2025 Co-Op: R&D Packaging

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

Research Intern - Biomedical AI for Precision Health

Hybrid 🏡🏢 04 November
R&D, Innovation
Washington, United States 🇺🇸 Massachusetts, United States 🇺🇸 California, 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 🇺🇸

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 🇨🇦

Signal and Power Integrity Engineer (RDSS Intern)

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

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

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

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

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

Mixed Signal Design Engineer - New College Grad 2025

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

2025 Applied Scientist Internship, Amazon University Talent Acquisition

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

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

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

2025 Applied Scientist Internship, Amazon University Talent Acquisition

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

Design Verification Engineer (RDSS Intern)

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

Research Intern - Mathematics

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

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

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

Research Intern - Multimodal AI Research

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

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

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

Mixed Signal Design Engineer (RDSS Intern)

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

Student / Intern in Oracle Labs

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

Student / Intern in Oracle Labs App Dev Projects

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

Amazon Science Fellow, JP Manga

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

Applied Scientist Intern, International Technology, 2025 Beijing

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

Speech Synthesis Intern

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

游戏AI研究员

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

Research Assistant - PFE 2025

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

Research Sciences Resident (AI for Good) Intern

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

Design Verification Intern

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

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

On-site 🏢 02 October
R&D, Innovation
Seattle, 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 🇬🇧

Silicon Validation Engineer (RDSS Intern)

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

Server Platform Validation Intern

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

Speech Synthesis Intern

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

Speech Synthesis Intern

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

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

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

Research Engineer (L4) - Member Lifecycle and Monetization

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

Student / Intern in Oracle Labs App Dev Projects

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

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 Scientist, Design Automation - New College Grad 2025

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

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

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

Quantum Researcher PhD (Intern) United States

Remote 🌎🌍🌏 New 🔥🔥
R&D, Innovation
California, United States 🇺🇸

Mechanical Process Engineer Intern, 2025 Internship, Shanghai, China

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

Research and Development Intern

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

Mixed Signal Analog Circuit Designer (RDSS Intern)

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

2025 ASIC Physical Design Engineer Intern, Annapurna Labs

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

2025 Mechanical/Electrical Engineering Internship

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

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

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

System Design Power Validation Engineer - New College Grad

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

Product Development Engineer (RDSS Intern)

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

Mixed Signal Design Engineer (RDSS Intern)

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

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, Frontier AI & Robotics

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

Research Intern - Applied Sciences Group (Computer Vision)

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

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

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

PhD Research Intern, Generalist Embodied Agents Research - Fall 2025

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

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

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

Research Intern - Formal Language Processing

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

2025 Applied Scientist Internship, Amazon University Talent Acquisition

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

Hardware Development Engineer, Annapurna Labs

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

Research Intern - Gray Systems Lab

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

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

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

Applied Research Intern - 2025

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

Intern Researcher in Computer Vision and Machine Learning

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

Research Intern - Applied Sciences Group (Computer Vision)

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

Service Process Innovation & Systems Support (INTERNSHIP)

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

Research Intern - Bio LLMs

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

Student / Intern in Oracle Labs

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

Research Assistant - PFE 2025

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

Research Intern

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

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

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

Research Sciences Intern (VISION)

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

ASIC Design Verification Engineer, Annapurna Labs

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

Research Intern - Microsoft Teams CMD Labs

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

GPU Firmware Engineer (RDSS Intern)

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

Share this job, spread the word!

Apply now

Similar jobs

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

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

Research Intern - MSR Inclusive Futures Team

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

Research Intern - AI-Assisted Game Creation

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