Gregory Thornton 03a1ee9269 Update Jax linesearch to behave more like Scipy

2021-05-26 12:49:56 +01:00

..

optimize

Update Jax linesearch to behave more like Scipy

2021-05-26 12:49:56 +01:00

sparse

added bicgstab to new jax repo

2021-02-18 18:01:28 -08:00

stats

Replace pow() with srqt() or square() where possible

2021-05-24 10:43:35 +01:00

__init__.py

Move most of the implementation of jax.scipy into jax._src.scipy.

2020-10-16 17:04:25 -04:00

linalg.py

Add support for select and select_range to jax.scipy.linalg.eigh_tridiagonal().

2021-05-05 21:36:57 -04:00

ndimage.py

Move jax.api to jax._src.api.

2021-04-13 09:43:24 -07:00

signal.py

Fix #4775 + additional fixes

2020-11-09 10:40:14 +05:30

special.py

Move jax.api to jax._src.api.

2021-04-13 09:43:24 -07:00