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

Amazon

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

Annapurna Labs builds custom Machine Learning accelerators that are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Neuron Compiler team is searching for compiler-skilled engineering talent to support the development and scaling of a compiler to enable the world's largest ML workloads to run performantly on these custom Annapurna systems.The Product: The AWS Machine Learning accelerators represent a pinnacle of AWS technologies, specifically designed for advancing AI capabilities. The Inferentia/Trainium chips specifically offer unparalleled ML inference and training performances. They are enabled through state-of-the-art software stack - the AWS Neuron Software Development Kit (SDK). This SDK comprises an ML compiler, runtime, and application framework, which seamlessly integrate into popular ML frameworks like PyTorch. AWS Neuron, running on Inferentia and Trainium, is trusted and used by leading customers such as Snap, Autodesk, and Amazon Alexa.The Team: Annapurna Labs was a startup company acquired by AWS in 2015, and is now fully integrated. If AWS is an infrastructure company, then think Annapurna Labs as the infrastructure provider of AWS. Our org covers multiple disciplines including silicon engineering, hardware design and verification, software, and operations. AWS Nitro, ENA, EFA, Graviton and F1 EC2 Instances, AWS Neuron, Inferentia and Trainium ML Accelerators, and in storage with scalable NVMe, are some of the products we have delivered over the last few years.Within this ecosystem, the Neuron Compiler team is developing a deep learning compiler stack that takes state of the art LLM and Vision models created in frameworks such as TensorFlow, PyTorch, and JAX, and makes them run performantly on our accelerators. The team is comprised of some of the brightest minds in the engineering, research, and product communities, focused on the ambitious goal of creating a toolchain that will provide a quantum leap in performance.You: As a PhD Machine Learning Compiler Engineer on the AWS Neuron Compiler team, you will be supporting the ground-up development and scaling of a compiler to handle the world's largest ML workloads. Architecting and implementing business-critical features, publish cutting-edge research, and contributing to a brilliant team of experienced engineers excites and challenges you. You will leverage your technical communications skill as a hands-on partner to AWS ML services teams and you will be involved in pre-silicon design, bringing new products/features to market, and many other exciting projects.A background in compiler development is strongly preferred. A background in Machine Learning and AI accelerators is preferred, but not required.In order to be considered for this role, candidates must be currently located or willing to relocate to Cupertino (preferred), Seattle, or Toronto.

Apply now

Similar jobs

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

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

2025 Applied Scientist Internship, Amazon University Talent Acquisition

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Speech Synthesis Intern

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

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

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

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

On-site 🏢 New 🔥🔥
R&D, Innovation
Seattle, United States 🇺🇸

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Research Intern - Microsoft Teams CMD Labs

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

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

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

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

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Silicon Design Intern (Jul - Dec 2025)

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Speech Synthesis Intern

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

Artificial General Intelligence (Audio/Speech/Language Processing and Large Multimodal Model Focus) RESEARCH INTERNSHIP

On-site 🏢 New 🔥🔥
R&D, Innovation
Bellevue, United States 🇺🇸

PhD Research Intern, Generalist Embodied Agents Research - Summer 2025

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

Hardware Development Engineer, Annapurna Labs

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

2025 ASIC Physical Design Engineer Intern, Annapurna Labs

On-site 🏢 03 October
R&D, Innovation
Austin, United States 🇺🇸 Cupertino, 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 Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Mechanical Engineer: Internship Opportunities - PhD - Redmond

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

2025 Applied Scientist Internship, Amazon University Talent Acquisition

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

Applied Scientist Intern, International Technology, 2025 Beijing

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

VLSI Physical Design Engineer - New College Grad 2025

On-site 🏢 New 🔥🔥
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 🇫🇷

Mechanical Process Engineer Intern, 2025 Internship, Shanghai, China

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

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

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

Research and Development Intern

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

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

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

Product Development Engineer (RDSS Intern)

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

Design Verification Intern

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

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

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Silicon Validation Engineer (RDSS Intern)

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

Amazon Science Fellow, JP Manga

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Vehicle Adaptation Intern - 2025

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

ASIC Design Engineer - New College Grad 2025

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

Power Architecture and Optimization Engineer – New College Grad 2025

On-site 🏢 24 September
R&D, Innovation
California, United States 🇺🇸 Texas, United States 🇺🇸

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

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 🇮🇹

Speech Synthesis Intern

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

System Design Power Validation Engineer - New College Grad

On-site 🏢 23 April
R&D, Innovation
Taipei, Taiwan 🇹🇼

Research Engineer (L4) - Member Lifecycle and Monetization

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

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 Assistant - PFE 2025

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

Design Engineer, Coherent High Speed Interconnect - New College Grad 2025

On-site 🏢 New 🔥🔥
R&D, Innovation
California, United States 🇺🇸 Massachusetts, United States 🇺🇸

GPU Power Architect - New College Grad 2025

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

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 🇺🇸

PhD Intern, Applied Research Scientist Retriever - Fall 2025

On-site 🏢 New 🔥🔥
R&D, Innovation
California, United States 🇺🇸

Intern Researcher in Computer Vision and Machine Learning

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

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 Sciences Intern (VISION)

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

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

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

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

On-site 🏢 18 April
R&D, Innovation
Beijing, China 🇨🇳

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

Mixed Signal Design Engineer (RDSS Intern)

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

Quantum Researcher PhD (Intern) United States

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

SRAM Circuit Design Engineer - New College Grad 2025

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

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

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

ASIC Design Verification Engineer, Annapurna Labs

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Student / Intern in Oracle Labs

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

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

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

2025 Applied Scientist Internship, Amazon University Talent Acquisition

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

Server Platform Validation Intern

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

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

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

Research Assistant - PFE 2025

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

2025 Applied Scientist Internship, Amazon University Talent Acquisition

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

GPU Firmware Engineer (RDSS Intern)

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

Research Intern - Applied Speech Research

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

ASIC Design Engineer - New College Grad 2025

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

Research Fellow

Hybrid 🏡🏢 New 🔥🔥
R&D, Innovation
Karnataka, India 🇮🇳

Perf Engineering Intern - 2025

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

Research Assistant - PFE 2025

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

Research Intern - Applied Sciences Group (Computer Vision)

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

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 Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

PhD Research Intern, Generalist Embodied Agents Research - Fall 2025

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

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

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

Mixed Signal Design Engineer (RDSS Intern)

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

Research Intern

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

Firmware Intern (Jul - Dec 2025)

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

Service Process Innovation & Systems Support (INTERNSHIP)

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

Student / Intern in Oracle Labs

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

Mixed Signal Analog Circuit Designer (RDSS Intern)

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

Process Innovation Intern

Hybrid 🏡🏢 New 🔥🔥
R&D, Innovation
Zagreb, Croatia 🇭🇷

Applied Scientist Intern, Global CoreJP, 2025 Beijing

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Research Intern - Quantum Computing

Hybrid 🏡🏢 New 🔥🔥
R&D, Innovation
California, United States 🇺🇸

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

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

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

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Mixed-Signal Circuit Design Engineer - New College Graduate

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

Share this job, spread the word!

Apply now See all roles in United States 🇺🇸

Similar jobs

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

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

Research Assistant OIP 2025 (Summer Internship)

On-site 🏢 25 April
R&D, Innovation
Casablanca, Morocco 🇲🇦

2025 Applied Scientist Internship, Amazon University Talent Acquisition

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