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