Senior ML Kernel Performance Engineer — AI Acceleration
Job in
Cupertino, Santa Clara County, California, 95014, USA
Listed on 2026-09-05
Listing for:
Annapurna Labs (U.S.) Inc.
Full Time
position Listed on 2026-09-05
Job specializations:
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer
Job Description & How to Apply Below
Annapurna Labs (U.S.) Inc. at Amazon Web Services (AWS) seeks a software engineer to design and optimize high-performance compute kernels for ML operations on the Neuron architecture.
You will leverage the Neuron architecture and programming models to push the limits of ML inference and training on Inferentia and Trainium. Based in Cupertino, you will analyze kernel-level performance, implement compiler optimizations such as fusion, sharding, tiling, and scheduling, and collaborate with
#J-18808-LjbffrPosition Requirements
10+ Years
work experience
To View & Apply for jobs on this site that accept applications from your location or country, tap the button below to make a Search.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).
Search for further Jobs Here:
×