3 ms·

Interesting. Any links to examples or docs on how to use PyTorch as a general linear algebra library for this purpose? Like a “SciPy to PyTorch” transition guid

by setopt 2y ago

Interesting. Any links to examples or docs on how to use PyTorch as a general linear algebra library for this purpose? Like a “SciPy to PyTorch” transition guide if I want to do the same?

meisel 2y ago
It's typically just importing torch and s/np/torch, not too different from NumPy -> CuPy. Try it in your own code and see!
ttyprintk 2y ago
Mentioned above: https://data-apis.org/array-api-compat/ https://data-apis.org/array-api-compat/