jax
ed3e2308 - Add support for linear algebra ops on GPU using Cusolver:

Commit
6 years ago
Add support for linear algebra ops on GPU using Cusolver: * LU decomposition * Symmetric (Hermitian) eigendecomposition * Singular value decomposition. Make LU decomposition tests less sensitive to the exact decomposition; check that we have a decomposition, not precisely the same one scipy returns.
Author
Parents
Loading