
Amazon
Join to apply for the ML Compiler Software Engineer PEY Co-op (12-16 months), Annapurna ML role at Amazon
Join to apply for the ML Compiler Software Engineer PEY Co-op (12-16 months), Annapurna ML role at Amazon
Get AI-powered advice on this job and more exclusive features.
Description
At AWS our vision is to make deep learning pervasive for everyday developers and to democratize access to cutting edge infrastructure. In order to deliver on that vision, we’ve created innovative software and hardware solutions that make it possible.
Description
At AWS our vision is to make deep learning pervasive for everyday developers and to democratize access to cutting edge infrastructure. In order to deliver on that vision, we’ve created innovative software and hardware solutions that make it possible.
AWS Neuron is the SDK that optimizes the performance of complex neural net models executed on AWS Inferentia and Trainium, our custom chips designed to accelerate deep-learning workloads.
The Neuron SDK consists of a compiler, run-time, and debugger, integrated with Tensorflow, PyTorch, and MXNet. It is preinstalled in AWS Deep Learning AMIs and Deep Learning Containers for customers to quickly get started with running high performance and cost-effective inference and training.
The Neuron team is looking for students interested in learning about and contributing to deep learning compilers to solve our customers toughest problems. As a compiler engineer on the Neuron team, you will be supporting the development of a compiler targeting AWS Inferentia and Trainium – our custom Silicon ML accelerators. This is an opportunity to work on a cutting-edge product where you will implement business-critical features, and be mentored by a brilliant team of experienced engineers.
Basic Qualifications
- Experience with at least one modern language such as Java, Python, C++, or C# including object-oriented design
- Currently enrolled in a Bachelor’s degree program or higher majoring in Computer Science, Computer Engineering, Electrical Engineering or related technical field with anticipated graduation date on or later than May 2026
- Candidates with strong interests and academic qualifications/research focus in two of the following: 1. Knowledge of code generation, compute graph optimization, resource scheduling, 2. Data structure and algorithms, 3. Compiler – Optimizing compilers (internals of LLVM, clang, etc), 4. Machine Learning – Experience with XLA, TVM, MLIR, LLVM, 5. Machine Learning frameworks (PyTorch, JAX) and Machine Learning models
- Available for 12-month internship starting May 2025
Preferred Qualifications
- Previous technical internship(s)
- Knowledge code generation, compute graph optimization, resource scheduling
- Experience in optimization mathematics such as linear programming and nonlinear optimization
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.
Company – Amazon Development Centre Canada ULC
Job ID: A2770360
Seniority level
Seniority level
Not Applicable
Employment type
Employment type
Full-time
Job function
Job function
Information Technology, Consulting, and EngineeringIndustries
Software Development
Referrals increase your chances of interviewing at Amazon by 2x
Get notified about new Software Engineer Intern jobs in Toronto, Ontario, Canada.
Software Developer Intern/Co-op, Backend (Fall 2025)
Software Development Intern (e-Transfer)
Co-Op/Intern, Software Development – Fall 2025
Software Engineering Intern 2025 (Toronto)
Software Development Intern (September – December 2025)
Amazon Robotics – Software Development Engineer Co-Op – 2025 – Toronto
Software Developer Intern – May 2026 (4 months)
Software Analyst Intern (12 or 16months)
Software Engineering Intern/Co-Op (Undergrad | Fall, 2025 | Onsite/Hybrid | 4 months)
Markham, Ontario, Canada $44,500.00-$71,000.00 2 months ago
Amazon Robotics – Software Development Engineer Co-op – 2025 – Toronto
Software Engineering Intern, Robotics Perception Research – Fall 2025
Graphics Software Engineering Intern/Co-op (Undergrad | Fall, 2025 | Onsite/Hybrid | 4 months)
Markham, Ontario, Canada $44,500.00-$71,000.00 2 months ago
Intern Research Engineer – Software Systems Engineering/LLMs
Software Test Engineering Intern/Co-Op (Undergrad | Fall, 2025 | Onsite/Hybrid | 4 months)
Markham, Ontario, Canada $44,500.00-$71,000.00 1 month ago
Spring 2026 Electronics Internship (BS/MS)
We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.
#J-18808-Ljbffr