Senior ML Kernel Optimizer AWS Neuron
Listed on 2026-10-06
-
IT/Tech
AWS, Cloud Computing: Infrastructure & Operations, Machine Learning/ ML Engineer, AI Engineer (Applied/Software)
Annapurna Labs (U.S.) Inc. seeks experienced kernel engineers to optimize ML workloads on AWS accelerators. You will design high-performance compute kernels for ML ops and work across compiler, runtime, framework, and hardware teams to maximize Neuron-based performance.
The role emphasizes low-level optimization, system architecture, and collaboration with customers to enable model acceleration on Inferentia and Trainium. Cupertino, CA-based team location with a strong learning culture.
This role, Senior ML Kernel Optimizer for AWS Neuron at Amazon Web Services (AWS), could be your next career step.
The Senior ML Kernel Optimizer for AWS Neuron role at Amazon Web Services (AWS) is now open for applications in Cupertino, CA, United States.
Join us at Amazon Web Services (AWS) as our next Senior ML Kernel Optimizer for AWS Neuron in Cupertino, CA, United States.
(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).