Senior ML Kernel Optimizer for AWS Neuron

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.