Peter Hawkins ed3e2308c1 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.

2019-08-02 11:16:15 -04:00

..

__init__.py

Add support for linear algebra ops on GPU using Cusolver:

2019-08-02 11:16:15 -04:00

xla_bridge.py

Replace jaxlib import paths with jax.lib paths.

2019-07-30 21:01:48 -04:00