Unknown Company

Senior ML Kernel Performance Engineer - AI Accelerator

cupertino, ca • Posted 2 weeks ago
Onsite Full Time IT & Technology

Amazon's Annapurna Labs seeks a Sr. ML Kernel Performance Engineer to advance AWS Neuron on Inferentia and Trainium. You will design high-performance kernels and optimize across generations, collaborating with compiler, runtime, framework, and hardware teams to push ML throughput.

This role offers exposure to cutting-edge AI acceleration and opportunities to mentor engineers while delivering production-grade kernel optimizations that scale with customer needs.

#J-18808-Ljbffr

Senior ML Kernel Performance Engineer - AI Accelerator in cupertino at Unknown Company

This position is listed as full time and onsite.

Back to Job Search