TryApplyNow

AWS Neuron ML Kernel Engineer

📍 Location
toronto, on
⏰ Job Type
Full-time
📅 Posted
June 01, 2026
Apply Now

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

Start Your Week Right!

Apply now and make every Monday exciting with TryApplyNow

Apply for this Position