Join AWS as a Machine Learning Kernel Engineer in Toronto, Ontario, where you'll develop high-performance kernels for advanced ML applications. Enhance model performance for AWS's unique ML accelerators while collaborating with innovative teams.
This junior engineering role at AWS focuses on kernel performance optimization for deep learning and generative AI workloads. Candidates must have over three years of software development experience and a strong understanding of kernel-level performance improvements. You will work hand-in-hand with various teams to innovate kernel designs tailored for customer ML projects.
Key Responsibilities:
• Develop high-performance computation kernels for machine learning applications
• Assess kernel performance across generations of Neuron hardware
• Execute performance profiling to identify and resolve bottlenecks
• Apply compiler optimizations through fusion and scheduling techniques
• Collaborate to refine customer machine learnin...