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 • Proficient in...