Machine Learning Kernel Engineer at AWS

Enhance deep learning performance at Amazon Web Services as a Machine Learning Kernel Engineer. Your expertise will focus on optimizing AWS Neuron for state-of-the-art hardware.
With the Annapurna Labs team, you'll work on critical projects to refine kernel performance for machine learning applications. This hands-on position invites collaboration with diverse teams and direct engagement with customers, aiming to improve their use of AWS technologies through advanced optimization techniques.
Key Responsibilities:
• Design compute kernels optimized for Neuron architecture
• Optimize kernel performance across various hardware platforms
• Analyze performance using profiling tools
• Implement techniques like fusion and scheduling for performance enhancement
• Collaborate effectively with internal teams and customer stakeholders
Requirements:
• Minimum of 3 years in qualified software development
• At least 2 years in systems design or architecture
• Profici...