Pytorch AMD XDNA2 (ryzen AI processors in new laptops) inference and training backend
Hi, I was bit annoyed about my XDNA2 npu being not really useful on my laptop so I developed pytorch backend that allows accelerated training and inference of pytorch on the NPU. Its around 1.3x-2x faster then regular CPU. If you want to try it You can check this out: https://github.com/mistrjirka/…
Read the full story at r/learnmachinelearning ↗
Timeline · 1 report
- 2026-10-14 12:55 · r/learnmachinelearning
Pytorch AMD XDNA2 (ryzen AI processors in new laptops) inference and training backend