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 innovative optimization techniques
• Enable and optimize customer ML models on AWS accelerators

Requirements:
• 5+ years of non-internship software development experience
• Solid 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 architectures

Drive exceptional performance and innovation in ML at Amazon.
#J-18808-Ljbffr

📌 Amazon ML Kernel Performance Engineer (Ontario)
🏢 Amazon Web Services (AWS)
📍 Ontario