1 day ago
Toronto, CanadaIntern
Responsibilities
- Bring up and optimize machine learning models for peak performance on AWS Trainium.
- Write and tune kernels, optimize sharding and model execution, and build benchmark and measurement infrastructure.
- Develop reusable Neuron components and optimization techniques for open-source and customer workloads.
- Build profiling, debugging, performance analysis, visualization, and developer workflow tools through the Neuron Explorer and Developer Experience team.
- Solve customer and engineering problems while collaborating with and learning from machine learning and systems engineers.
Requirements
- 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++.
- Academic, research, or project experience in at least two areas including performance engineering, profiling, benchmarking, low-level systems optimization, kernel development, parallel programming, computer architecture, developer tooling, data structures and algorithms, machine learning frameworks and models, or compiler and ML systems technologies.
- Preferred experience includes optimizing machine learning models, writing or optimizing kernels for GPUs, ML accelerators, or FPGAs, using performance analysis tools, building developer tools, technical internships or relevant research, full-stack development, TypeScript, React, Go, and agentic workflows.
- Ability to explain technical challenges and solutions clearly and work independently through ambiguous or undefined problems.
Benefits
- 12–16 month internship starting May 2027 or 3–4 month internship starting January, May, or September 2027.
- The position includes mentorship from experienced machine learning and systems engineers.
- Amazon offers basic life and AD&D insurance, paid time off, and other health and well-being resources.
- The position is based in Toronto, Ontario, Canada.
Categories
About Amazon
Amazon builds and operates a global e-commerce marketplace, logistics network, and consumer devices, and provides cloud computing via AWS for businesses and developers. The company earns revenue from online retail, third‑party seller services, subscriptions like Prime, advertising, and AWS usage. Founded in 1994 and headquartered in Seattle, it is publicly traded on NASDAQ (AMZN) and serves customers in dozens of countries.