Location
toronto, on, Canada
Posted
June 01, 2026
Job 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 • Knowledgeable in at least one programming language
Utilize y...
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 • Knowledgeable in at least one programming language
Utilize y...