Position Overview
Enhance machine learning performance as a Kernel Performance Engineer at Annapurna Labs, AWS. Focus on developing high-performance compute kernels for leading-edge ML accelerators like Inferentia and Trainium.
This position seeks candidates with over 3 years of software development exposure, ideally targeting optimized ML functions. You'll analyze kernel performances, work cross-functionally on innovative solutions, and facilitate customer ML model performance on AWS platforms.
Key Responsibilities:
β’ Design and implement kernels for machine learning applications
β’ Optimize kernel performance across different hardware
β’ Conduct profiling and analysis to address issues
β’ Apply compiler optimizations for improved efficiency
β’ Collaborate with various teams to drive technical innovations
Requirements:
β’ 3+ years in professional software development
β’ 2+ years in system design or architecture
β’ Knowledge...