A

AWS Neuron Kernel Optimization Engineer

Amazon

toronto, on, Canada Full-time May 29, 2026

Opportunity Description

Enhance machine learning performance as a Kernel Performance Engineer at Annapurna Labs, AWS. Focus on developing high-performance compute kernels for leading-edge ML accelerators like Inferentia and Trainium.

This position seeks candidates with over 3 years of software development exposure, ideally targeting optimized ML functions. You'll analyze kernel performances, work cross-functionally on innovative solutions, and facilitate customer ML model performance on AWS platforms.

Key Responsibilities:
• Design and implement kernels for machine learning applications
• Optimize kernel performance across different hardware
• Conduct profiling and analysis to address issues
• Apply compiler optimizations for improved efficiency
• Collaborate with various teams to drive technical innovations

Requirements:
• 3+ years in professional software development
• 2+ years in system design or architecture
• Knowledge...
Full-time IT & Technology

Interested in this opportunity? Apply now through Expertini.

Apply for this Position