Unknown Company

Senior ML Kernel Performance Architect

cupertino, ca • Posted 5 days ago
Onsite Full Time Engineering

Amazon is seeking a Sr. ML Kernel Performance Engineer for AWS Neuron at Annapurna Labs to design and optimize high‑performance kernels for ML operations on Neuron accelerators. You will collaborate with hardware, compiler, runtime, and framework teams to maximize throughput and efficiency.

Responsibilities include profiling, identifying bottlenecks, implementing fusion, tiling, and scheduling, and enabling customers to deploy optimized models on AWS accelerators.

#J-18808-Ljbffr
Back to Job Search