Full-time

Amazon Senior ML Performance Engineer

Posted by Amazon • toronto, on, Canada

📍 toronto, on 🕒 June 01, 2026

About the Role

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

Ready to Apply?

Submit your application today and take the next step in your career journey with Amazon.

Apply Now