🎓 JobsStudent.com

Your Gateway to Student Success

← Back to Student Jobs

Senior ML Kernel Performance Engineer

Student-Friendly Entry-Level
Company

Amazon

Location

Toronto, Canada

Posted

June 02, 2026

🎓 Ready to Start?

Join thousands of students launching their careers

Apply Now

About This Opportunity

Description
The Annapurna Labs team at Amazon builds Neuron, the software development kit used to accelerate deep learning and GenAI workloads on Amazon’s custom machine learning accelerators, Inferentia and Trainium.

The Acceleration Kernel Library team is at the forefront of maximizing performance for Amazon's custom ML accelerators. Working at the hardware-software boundary, our engineers craft high-performance kernels for ML functions, ensuring every FLOP counts in delivering optimal performance for our customers' demanding workloads. We combine deep hardware knowledge with ML expertise to push the boundaries of what's possible in AI acceleration.

The Amazon Neuron SDK, developed by the Annapurna Labs team at Amazon, is the backbone for accelerating deep learning and GenAI workloads on Amazon's Inferentia and Trainium ML accelerators. This comprehensive toolkit includes an ML compiler, runtime, and application framework that seamlessly integrates with popular M...