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 operations
Analyze and optimize kernel-level performance across generations




Utilize profiling tools for performance analysis
Collaborate to develop cutting-edge optimization techniques
Enable and optimize customer ML models on AWS acceleratorsRequirements:
5+ years of non-internship software development experience
Robust programming skills in at least one language
5+ years of leading design or architecture
Mentorship experience or tech lead background
In-depth knowledge of ML accelerator architecturesDrive exceptional performance and innovation in ML at Amazon.#J-18808-Ljbffr

📌 Amazon Ml Kernel Performance Engineer Toronto
🏢 Amazon Web Services (AWS)
📍 Toronto