Junior ML Kernel Engineer at Amazon

Join AWS as a Junior Machine Learning Kernel Engineer based in Toronto, Ontario. Specialize in creating high-performance kernels for proprietary ML accelerators like Inferentia and Trainium.
Become a part of Annapurna Labs at Amazon Web Services, where your work will directly impact deep learning and generative AI workloads. This junior role requires over three years of software development experience with a significant focus on kernel-level performance optimizations. Collaborate with cross-functional teams to enhance kernel designs for client ML models.
Key Responsibilities:
• Create high-performance kernels for various machine learning tasks
• Evaluate kernel performance across different Neuron hardware versions
• Conduct thorough performance profiling to uncover bottlenecks
• Implement key compiler optimizations like fusion and scheduling
• Partner with teams to fine-tune customer ML model performance
Requirements:
• At least 3 years in software developmen...