Unknown Company

ML Kernel Performance Engineer for Neuron Accelerators

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

Amazon's Annapurna Labs in the AWS Neuron group seeks a dedicated ML Kernel Performance Engineer to design and optimize high-performance compute kernels for ML workloads on Inferentia and Trainium. You will push kernel optimization boundaries across generations and collaborate with cross-functional teams.

This role partners directly with customers to tune models for AWS accelerators, applies compiler optimizations, and participates in design reviews and code quality activities in a startup-like

#J-18808-Ljbffr

ML Kernel Performance Engineer for Neuron Accelerators in cupertino at Unknown Company

This position is listed as full time and onsite.

Back to Job Search