Improving Speed with CUDA

[wert93]

Is there a way to improve the speed performance of the solver by doing the computation on GPU? If so, which tensors should be assigned to 'cuda'?

[Roythuly]

I find if you use your CPU with this repo, the speed performance would be improved well