jobs

Machine Learning Performance Engineer, Annapurna Labs

The Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and generative AI workloads on Amazon's custom machine learning accelerators — Inferentia and Trainium. Th

Company

Amazon

Location

Tel Aviv-Yafo, Tel Aviv, ISR

Status

Open

Posted

2026-09-22T00:00:00+00:00

Amazon's Annapurna Labs is hiring a machine learning performance engineer for a new Neuron performance team in Tel Aviv. The role involves profiling ML workloads, writing high-performance kernels, improving the Neuron SDK, and working with compiler, runtime, and chip engineers to remove bottlenecks. The posting asks for 3+ years of software development, Python or C++, computer architecture, parallel computing, and PyTorch, TensorFlow, or JAX. Preferred experience includes kernel writing, LLM optimization, and compiler or hardware-software co-design. The team is newly formed.