Performance of Pytorch backend vs CPU

[RishalAggarwal]

I see now several OT algorithms have pytorch backend support, I'm wondering if using the pytorch backend on GPU speeds up the convergence of these solvers?

[rflamary]

Yes it should be faster on GPU. Still some solvers (exact OT, emd, non regularized GW) use a CPU C++ solvers and those ones will work with GPU array but incur a memory copy overhead.