Elevate your career with Amazon as a Machine Learning Kernel Performance Engineer, focusing on high-performance kernels. Join the Annapurna Labs team to shape the future of AI acceleration technology.In this senior role, contribute your expertise in ML model acceleration while optimizing performance on Amazon’s custom hardware, Inferentia and Trainium. You'll work at the intersection of software and hardware, designing cutting-edge ML solutions and collaborating with various teams.



Your ability to analyze performance bottlenecks and implement optimizations will be crucial in delivering exceptional user experiences.Key Responsibilities:Craft high-performance compute kernels for ML operationsAnalyze and optimize kernel-level performance across generationsUtilize profiling tools for performance analysisCollaborate to develop cutting-edge optimization techniquesEnable and optimize customer ML models on AWS acceleratorsRequirements:5+ years of non-internship software development experienceStrong programming skills in at least one language5+ years of leading design or architectureMentorship experience or tech lead backgroundIn-depth knowledge of ML accelerator architecturesDrive exceptional performance and innovation in ML at Amazon.#J-18808-Ljbffr

📌 Amazon Ml Kernel Performance Engineer (Toronto)
🏢 Amazon
📍 Toronto