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 ...