Amazon ML Kernel Performance Engineer
2 days ago
Toronto ON, Toronto Census Division, ON; Ontario, Canada
Amazon Web Services (AWS)
Full-time
Free with email or Google
Save this job and keep your search organized
Create a free account to save jobs, create alerts and return to this listing from your dashboard.
Free with email or Google
By continuing, you agree to our Terms & Privacy Policy.
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
• Strong 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
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
• Strong 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