Skip to content

ML Compiler Software Engineer PEY Co-op (12-16 months), Annapurna ML

  • Full Time
  • Internship
  • Toronto

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 Engineering
  • Industries

    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

To apply, please visit the following URL:

THISJOB.CA