You can run pyTorch without GPU acceleration, the API could be useful for your problem even if you won’t get any super-speedup. It certainly will be faster than python loops even without the GPU, if pytorch is compiled with Intel MKL support etc. (I’m just guessing, I noticed that it uses MKL but I haven’t done any real pure CPU benchmarks)