Join the frontier of AI acceleration as a Senior ML Performance Engineer at Amazon's Annapurna Labs. Help advance deep learning and GenAI workloads on advanced machine learning accelerators.

You will architect high-performance kernels and optimize performance at the hardware-software boundary in this impactful role. Drive innovation in ML techniques, analyze kernel performance, and directly support customers with their AWS model optimization needs. Collaborate cross-functionally while operating in a fast-paced, innovative environment.

Key Responsibilities: • Implement high-performance compute kernels leveraging Neuron architecture • Optimize kernel-level performance across multiple hardware versions • Conduct performance analysis using various profiling tools • Collaborate on innovative kernel optimization strategies • Enable and support customers in their ML model performance

Requirements: • 5+ years professional software development experience • Mastery in ...