ML Systems Software Development Engineer Intern, Annapurna Labs - 2027

At Amazon, our tech isn't just a tool, it's your playground. Our engineers work on scalable systems, cloud services, and customer-facing products that operate at global scale. This is an environment where you learn by building. Whether you're creating cloud-native solutions, optimizing machine learning models, or building products that drive experiences for millions of customers around the globe, your work reaches the world, fast. You'll take ownership early, collaborate across teams and disciplines, and grow your skills as you tackle new problems. There's no single path forward here, with a chance to explore different directions and shape your journey as you go. This is where ambition meets opportunity and, where the impact of what you build helps define what comes next, for customers and for you.

What's in it for you? An internship at Amazon means real responsibility from day one. You'll build, test, and learn alongside people who want to see you succeed, while making an impact that reaches far beyond campus.

Key job responsibilities Join the Toronto Neuron team (Annapurna Labs) to bring up and optimize state-of-the-art machine learning models for peak performance on AWS Trainium and build industry-leading profiling tools that help engineers identify and fix bottlenecks.

The Toronto Neuron team is hiring 2027 interns in two focus areas:

• Frontier Model Performance team. Take newly released machine learning models

from first bring-up to peak performance on current and next-generation AWS Trainium silicon. You could write and tune kernels, optimize sharding and model execution, build benchmark and measurement infrastructure, or tackle architectural bottlenecks that shape future Trainium designs. The best ideas do not stay trapped in one model. The team turns them into reusable Neuron components and optimization techniques for flagship open-source and customer workloads.

• Developer Experience (DevEx) team. Make AWS Trainium performance visible. The

team owns Ne