Create Alert
Email me similar jobs

Senior ML Kernel Optimization Engineer

Full-time Kernel Ml

Annapurna Labs (U.S.) Inc. in Cupertino, CA is seeking a kernel engineer to design high-performance ML compute kernels for Neuron hardware and collaborate across compiler, runtime, and hardware teams to accelerate ML workloads on AWS accelerators.

You will optimize kernel performance across Neuron generations, perform profiling, and implement fusion, tiling, and scheduling while working with customers to tailor ML models for AWS accelerators.

#J-18808-Ljbffr
Similar jobs

Senior ML Kernel Optimization Engineer

Apply Now
Back to search page