Developer Experience (DevEx) team. Make AWS Trainium performance visible. The team owns Neuron Explorer and builds state-of-the-art profiling, debugging, and analysis tools that let engineers see how efficiently models and kernels use the hardware, pinpoint bottlenecks, and decide what to optimize next. You could build low-level performance data collection, analysis engines, interactive visualizations, or developer workflows. The goal is to cut through complex execution data so customers can debug faster and reach higher performance on Trainium.
Interns will join one of these teams based on their interests and experience. In either role, you will solve real customer and engineering problems with mentorship from experienced machine learning and systems engineers.
Internship Options
12-16 month internship (starts May 2027)
3-4 month internship (starts January 2027, May 2027, September 2027)
About the team
Annapurna Labs designs the custom silicon at Amazon - including Graviton (our server processors), Trainium and Inferentia (our machine learning training and inference accelerators), and the Nitro system that powers modern EC2. We own the full stack, from silicon through the software that makes it work (compilers, runtimes, drivers, and the frameworks that let customers run workloads on our hardware). As a Software Development Engineer Intern, you'll write the software that turns custom silicon into products used by millions.
Currently enrolled in a Bachelor's degree program or higher in Computer Science, Computer Engineering, Electrical Engineering, or a related field
Programming experience through coursework, research, or a previous internship using Python, C, and/or C++
Strong interest and academic, research, or project experience in at least two of the following areas: 1/ Performance engineering, profiling, benchmarking, or low-level systems optimization. 2/ Kernel development, parallel programming, or computer architecture. 3/ Developer tooling, including profilers, debuggers, diagnostics, or visualization. 4/ Data structures and algorithms. 5/ Machine learning frameworks and models, including PyTorch or JAX. 6/ Compiler or ML systems technologies such as LLVM, MLIR, XLA, or TVM
Experience bringing up and optimizing machine learning models
Experience writing or optimizing kernels for GPUs, ML accelerators, or FPGAs
Experience using performance analysis tools or building developer tools
Previous technical internship or relevant research experience
Experience with full stack development, including front-end technologies such as TypeScript and React, and back-end services (experience with Go is a plus)
Ability to explain technical challenges and solutions clearly
Ability to independently work through ambiguous or undefined problems and think abstractly
Experience in developing agentic workflows for work automation
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.
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. In addition, Amazon offers basic life & AD&D insurance, paid time off, and other resources to improve health and well-being. We thank all applicants for their interest, however only those interviewed will be advised as to hiring status.
Toronto, ON, CAN - 100,810.00 CAD Annually
