Join Amazon's Annapurna Labs as a Senior ML Kernel Performance Engineer and influence the future of AI acceleration. Focus on high-performance kernel design and delivery of cutting-edge optimization strategies for deep learning workloads. In this pivotal role, you will work with the Neuron SDK, enhancing its capabilities for Amazon's unique ML accelerators.

Collaboratively engage with teams across various layers of technology, balancing creative solutions with hands-on support for customer ML model optimization. Your expertise will help drive the efficiency of complex systems and architecture. Key Responsibilities:
Craft high-performance kernels for advanced ML applications
Evaluate and improve kernel-level performance




Leverage profiling tools for thorough performance analyses
Implement optimizations including tiling and scheduling
Collaborate with customers to enhance ML models on AWS Requirements:
Minimum 5 years in software development
At least 5 years of programming expertise
Proven experience in system design and architecture
Experience as a tech lead or mentor
Advanced knowledge of accelerator architectures beneficial Contribute your skills in software engineering and machine learning optimization at Amazon, helping customers achieve unprecedented performance.

📌 Senior Ml Kernel Performance Engineer Role Toronto
🏢 Amazon Web Services (AWS)
📍 Toronto