ML Kernel Performance Engineer, AI Accelerator

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