🌍 Global Opportunities
⚑ Updated Hourly
πŸŽ“ Student Friendly
⏰

parttimejobs.work

Flexible Work, Better Balance

⏰ Full-time

Senior ML Kernel Performance Engineer

Amazon
Location πŸ“ Toronto, Canada
Posted πŸ“… June 02, 2026
Work Type ⏰ Full-time

Position Overview

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...

Apply Now

Submit Application β†’

Quick and easy application process

Job Details

⏰
Employment Type
Full-time
πŸ“Š
Category
other-general
🏠
Work Arrangement
On-site
πŸ“
Location
Toronto, Canada