ML Kernel Performance Engineer, AWS Neuron, Annapurna Labs

Job description

Description The Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and GenAI workloads on Amazon's custom machine learning accelerators, Inferentia and Trainium. The Acceleration Kernel Library team is at the forefront of maximizing performance for AWS's custom ML accelerators. Working at the hardware-software boundary, our engineers craft high-performance kernels for ML functions, ensuring every FLOP counts in deliv…

Full details and the application form are on the source listing. Listing provided via Adzuna; posting details belong to the employer.