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 cutting-edge 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. #J-18808-Ljbffr

📌 Senior ML Kernel Performance Engineer Role (Winnipeg)
🏢 Amazon Web Services (AWS)
📍 Winnipeg