Job Description
Become an integral part of AWS as a Machine Learning Kernel Engineer in Toronto, Ontario. Focus on developing high-performance kernels tailored for AWS’s custom ML accelerators, enhancing ML model performance.
Work with the Annapurna Labs team at Amazon Web Services, focusing on kernel performance optimization for deep learning and generative AI workloads. As a junior engineer, you will need over three years of experience in software development and a solid grasp of kernel-level performance enhancements. Collaborate with various teams to innovate kernel designs for customer ML projects.
Key Responsibilities:
• Develop high-performance computation kernels for ML applications
• Assess and analyze kernel performance across generations of Neuron hardware
• Execute performance profiling to discover and resolve bottlenecks
• Apply compiler optimizations through fusion and scheduling techniques
• Work alongside teams to refine customer ML models
Requirements:
• 3+ y...
Work with the Annapurna Labs team at Amazon Web Services, focusing on kernel performance optimization for deep learning and generative AI workloads. As a junior engineer, you will need over three years of experience in software development and a solid grasp of kernel-level performance enhancements. Collaborate with various teams to innovate kernel designs for customer ML projects.
Key Responsibilities:
• Develop high-performance computation kernels for ML applications
• Assess and analyze kernel performance across generations of Neuron hardware
• Execute performance profiling to discover and resolve bottlenecks
• Apply compiler optimizations through fusion and scheduling techniques
• Work alongside teams to refine customer ML models
Requirements:
• 3+ y...