Unknown Company

ML Kernel Performance Engineer for AI Accelerators

cupertino, ca • Posted 4 days ago
Onsite Full Time IT & Technology

Annapurna Labs (U.S.) Inc. in Cupertino, CA, is seeking kernel engineers to optimize ML workloads on AWS accelerators.

You will design and implement high-performance kernels, analyze kernel-level performance, and collaborate across hardware, software, and ML teams to push the performance of AWS Neuron for Inferentia and Trainium. Join a team that blends deep hardware knowledge with ML expertise to advance AI acceleration, publish cutting-edge research, and mentor engineers in a startup-like,

#J-18808-Ljbffr
Back to Job Search