T

AWS Neuron ML Kernel Engineer

TryApplyNow

toronto, on, Canada Full-time June 01, 2026

Opportunity Description

# AWS Neuron ML Kernel EngineerAmazonFull TimejuniorToronto, Ontario, CAPosted 2 days ago## Job DescriptionJoin the Annapurna Labs team at Amazon Web Services as a Machine Learning Kernel Performance Engineer. Focus on designing high-performance kernels for AWS's custom ML accelerators, such as Inferentia and Trainium. As part of the Acceleration Kernel Library team, you'll work on maximizing performance for deep learning and generative AI workloads.This role requires 3+ years of software development experience and expertise in kernel-level optimizations. You will collaborate with various teams to innovate and enhance kernel design for customer ML models. Key Responsibilities:* Design high-performance compute kernels for ML functions* Analyze kernel performance across Neuron hardware generations* Conduct performance profiling to identify bottlenecks* Implement compiler optimizations like fusion and scheduling* Collaborate with teams to optimize customer ML models Requirements:* 3+ year...
Full-time Other-General

Interested in this opportunity? Apply now through Expertini.

Apply for this Position