ML Kernel Performance Engineer, AI Accelerator
15 hours ago
Toronto, ON, Canada
Amazon Inc.
Full-time
Free with email or Google
Save this job and keep your search organized
Create a free account to save jobs, create alerts and return to this listing from your dashboard.
Free with email or Google
By continuing, you agree to our Terms & Privacy Policy.
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