Senior ML Kernel Performance Engineer

Is this the right opportunity for you?

Explore more jobs, compare advertised salaries and see which skills employers want.

Search for a different role

Description

The Annapurna Labs team at Amazon builds 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 Amazon's custom ML accelerators.

Working at the hardware-software boundary, our engineers craft high-performance kernels for ML functions, ensuring every FLOP counts in delivering optimal performance for ou…

Get similar jobs in Canada by email

We'll email you when new jobs similar to this one appear.

Similar jobs

Explore more Software Development jobs in Canada.

Finding similar jobs…