Peter Hawkins d658108d36 Fix type errors with current mypy and NumPy. ...

Enable type stubs for jaxlib.

Fix a nondeterminism problem in jax2tf tests.

2021-06-24 10:51:06 -04:00

..

optimize

Fix type errors with current mypy and NumPy.

2021-06-24 10:51:06 -04: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

Support complex numbers in jax.scipy.signal.convolve/correlate

2021-06-18 13:07:00 -06:00

special.py

Support value computation of associated Legendre functions.

2021-06-14 14:51:37 -07:00