ML Kernel Performance Engineer, AI Accelerator

15 hours ago

Toronto, ON, Canada Amazon Inc. Full-time
Amazon Web Services (AWS) Annapurna Labs seeks a ML Kernel Performance Engineer for AWS Neuron, to push high-performance kernel development for Inferentia and Trainium. You will design and optimize kernels, collaborate across compiler, runtime, framework, and hardware teams, and work directly with customers to maximize ML workloads on AWS accelerators, shaping the future of AI acceleration technology. Join a small, agile team within a larger organization, enjoy a learning culture, publish