2026 ML Compiler Software Engineer PEY Co-op (12-16 months), AWS Neuron, Annapurna Labs
Share
Job Location:
Monthly Salary:
Posted on:
30+ days ago
Vacancies:
1 Vacancy
Department:
Job Summary
At AWS our mission is to make deep learning accessible to every developer by democratizing access to cutting-edge infrastructure. To achieve this weve built custom silicon (AWS Inferentia and Trainium) and the AWS Neuron SDK that together deliver high-performance cost-effective machine learning in the cloud.
The AWS Neuron SDK includes a compiler runtime debugger and libraries integrated with popular frameworks such as PyTorch and TensorFlow. It is preinstalled in AWS Deep Learning AMIs and Containers so customers can quickly get started with training and inference on AWS ML accelerators.
The Neuron Toronto team focuses on performance kernels and toolinganalyzing and optimizing end-to-end ML workloads developing and maintaining highly optimized kernels and building performance modeling profiling and accuracy debugging tools. Together these efforts ensure that Neuron delivers best-in-class performance flexibility and usability for customers deploying large-scale machine learning models.
As a student intern you will contribute to the efforts that make Neuron best-in-class for ML workloads. Youll gain hands-on experience building business-critical features analyzing performance developing compiler or kernel optimizations and building tools that provide deep insights into model execution. Youll be mentored by experienced engineers while working on technology that directly accelerates customer workloads at scale.
- Are enrolled in a academic program that is physically located in Canada
- Are enrolled in a Bachelors degree or higher in Computer Science Engineering Science Computer Engineering Electrical Engineering or majors relating to these fields with an anticipated graduation date between May 2027 - May 2028
- Strong interests and academic qualifications/research focus in two of the following: 1. Knowledge of code generation compute graph optimization resource scheduling 2. Compiler - Optimizing compilers (internals of LLVM clang etc) 3. Machine Learning frameworks (PyTorch JAX) and Machine Learning (Experience with XLA TVM MLIR LLVM) 4. Kernel developmentexperience writing CUDA kernels OpenCL kernels or ML-specific kernels
- Available for a 12-16-month internship starting May 2026
- Previous technical internship(s) related to the areas of interest / research focus listed above
- Experience in optimization mathematics such as linear programming and nonlinear optimization
- Academic coursework in Compiler Design/Construction Programming Language Theory Computer Architecture Advanced Algorithms & Data Structures
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 for more information. If the country/region youre applying in isnt listed please contact your Recruiting Partner.
The AWS Neuron SDK includes a compiler runtime debugger and libraries integrated with popular frameworks such as PyTorch and TensorFlow. It is preinstalled in AWS Deep Learning AMIs and Containers so customers can quickly get started with training and inference on AWS ML accelerators.
The Neuron Toronto team focuses on performance kernels and toolinganalyzing and optimizing end-to-end ML workloads developing and maintaining highly optimized kernels and building performance modeling profiling and accuracy debugging tools. Together these efforts ensure that Neuron delivers best-in-class performance flexibility and usability for customers deploying large-scale machine learning models.
As a student intern you will contribute to the efforts that make Neuron best-in-class for ML workloads. Youll gain hands-on experience building business-critical features analyzing performance developing compiler or kernel optimizations and building tools that provide deep insights into model execution. Youll be mentored by experienced engineers while working on technology that directly accelerates customer workloads at scale.
- Are enrolled in a academic program that is physically located in Canada
- Are enrolled in a Bachelors degree or higher in Computer Science Engineering Science Computer Engineering Electrical Engineering or majors relating to these fields with an anticipated graduation date between May 2027 - May 2028
- Strong interests and academic qualifications/research focus in two of the following: 1. Knowledge of code generation compute graph optimization resource scheduling 2. Compiler - Optimizing compilers (internals of LLVM clang etc) 3. Machine Learning frameworks (PyTorch JAX) and Machine Learning (Experience with XLA TVM MLIR LLVM) 4. Kernel developmentexperience writing CUDA kernels OpenCL kernels or ML-specific kernels
- Available for a 12-16-month internship starting May 2026
- Previous technical internship(s) related to the areas of interest / research focus listed above
- Experience in optimization mathematics such as linear programming and nonlinear optimization
- Academic coursework in Compiler Design/Construction Programming Language Theory Computer Architecture Advanced Algorithms & Data Structures
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 for more information. If the country/region youre applying in isnt listed please contact your Recruiting Partner.
Key Skills
- Spring
- .NET
- C/C++
- Go
- React
- OOP
- C#
- Data Structures
- JavaScript
- Software Development
- Java
- Distributed Systems
About Company
AmaZone Cafe it is a very cosy place, focus on healthy food, where you can delight with your friends and family and try the best Acai, the famous brazillian fruit.
