A

Amazon Senior ML Performance Engineer

Amazon

toronto, on, Canada Full-time May 31, 2026

Opportunity Description

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 profession...
Full-time Engineering

Interested in this opportunity? Apply now through Expertini.

Apply for this Position