[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 ASIC Physical Design Engineer Intern, Annapurna Labs

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

Research Intern - Biomedical AI for Precision Health

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

Student / Intern in Oracle Labs

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

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

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

Research Intern - Office of the Chief Scientific Officer

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

Mixed Signal Analog Circuit Designer (RDSS Intern)

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

SRAM Circuit Design Engineer - New College Grad 2025

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

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

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

GPU Firmware Engineer (RDSS Intern)

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

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

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

2025 Applied Scientist Internship, Amazon University Talent Acquisition

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

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

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

Student / Intern in Oracle Labs

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

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

Physical Design Backend Student

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

Research Intern - MSR Inclusive Futures Team

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

Research Intern - Gray Systems Lab

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

Intern Researcher in Computer Vision and Machine Learning

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

Student / Intern in Oracle Labs App Dev Projects

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

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

Student / Intern in Oracle Labs App Dev Projects

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

Research Scientist, Design Automation - New College Grad 2025

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

Research Intern - Applied Sciences Group (Computer Vision)

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

Server Platform Validation Intern

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

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

On-site 🏢 20 March
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 🇺🇸

Research Intern - Microsoft Teams CMD Labs

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

Research Intern - Multimodal AI Research

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

Mixed Signal Design Engineer (RDSS Intern)

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

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

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

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

On-site 🏢 15 November
R&D, Innovation
San Francisco, 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 🇺🇸

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

Speech Synthesis Intern

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

Cambridge Research Intern - MSR Intelligent Networked Systems

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

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

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

Research Intern - High performance and Scalable Database Systems

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

Firmware Intern (Jul - Dec 2025)

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

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

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

Mixed-Signal Circuit Design Engineer - New College Graduate

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

Research Intern - Mathematics

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

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

2025 Applied Scientist Internship, Amazon University Talent Acquisition

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

Signal and Power Integrity Engineer (RDSS Intern)

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

Student / Intern in Oracle Labs

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

PhD Research Intern, Generalist Embodied Agents Research - Summer 2025

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

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

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

Applied Research Intern - 2025

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

Speech Synthesis Intern

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

Mixed Signal Design Engineer (RDSS Intern)

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

Student / Intern_Oracle Labs

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

Vehicle Adaptation Intern - 2025

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

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

Silicon Validation Engineer (RDSS Intern)

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

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

Silicon Design Intern (Jul - Dec 2025)

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

Research Assistant - PFE 2025

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

Research Assistant - PFE 2025

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

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

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

Speech Synthesis Intern

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

Perf Engineering Intern - 2025

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

Research Sciences Intern (VISION)

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

Research Assistant - PFE 2025

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

Mixed Signal Design Engineer - New College Grad 2025

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

Design Verification Engineer (RDSS Intern)

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

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 Science Internship - Gen AI & Large Language Models - United States, PhD Student Science Recruiting

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

Service Process Innovation & Systems Support (INTERNSHIP)

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

Hardware Development Engineer, Annapurna Labs

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

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

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

Research Intern - Quantum Computing

Hybrid 🏡🏢 26 February
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 Intern - AI HW/SW Co-design

Hybrid 🏡🏢 10 January
R&D, Innovation
Washington, United States 🇺🇸 California, 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 - Applied Sciences Group (Computer Vision)

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

Quantum Researcher PhD (Intern) United States

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

Research Intern

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

Research Intern - Formal Language Processing

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

Research Intern - Applied Speech Research

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

游戏AI研究员

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

PhD Research Intern, Generalist Embodied Agents Research - Fall 2025

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

Amazon Science Fellow, JP Manga

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

Research and Development Intern

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

Design Verification Intern

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

GPU Power Architect - New College Grad 2025

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

Applied Scientist Intern, Global CoreJP, 2025 Beijing

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

2025 Applied Scientist Internship, Amazon University Talent Acquisition

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

2025 Mechanical/Electrical Engineering Internship

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

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

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

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

Applied Scientist Intern, International Technology, 2025 Beijing

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

Research Sciences Resident (AI for Good) Intern

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

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

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

Mechanical Process Engineer Intern, 2025 Internship, Shanghai, China

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

Share this job, spread the word!

Apply now

Similar jobs

2025 ASIC Physical Design Engineer Intern, Annapurna Labs

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

Research Intern - Biomedical AI for Precision Health

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

Student / Intern in Oracle Labs

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