Senior Performance Kernel Engineer at AWS

Shape the future of AI as a Senior Performance Kernel Engineer at AWS, part of Amazon’s Annapurna Labs. Focus on accelerating deep learning and GenAI workloads on cutting-edge accelerators.
In this role, you will design and implement high-performance kernels, perform detailed performance analyses, and employ advanced optimization techniques for Amazon’s machine learning accelerators. Engage directly with customers to support their model needs, driving collaboration and innovation at the cutting edge of technology.
Key Responsibilities:
• Design performance-enhancing compute kernels for ML operations
• Optimize performance using detailed analyses and profiling tools
• Implement sophisticated compiler optimizations and techniques
• Collaborate across teams to innovate kernel optimization
• Support customers in optimizing their machine learning models
Requirements:
• 5+ years in software development
• Proficient in a relevant programming language
• Experti...