jax authors 149a744892 Merge pull request #5533 from apaszke:general-collectives ...

PiperOrigin-RevId: 354942612

2021-02-01 08:15:32 -08:00

..

optimize

Minor doc formatting fixes

2021-01-29 16:43:27 -08:00

sparse

[JAX] Move jax.util to jax._src_util.

2021-01-11 14:21:07 -08:00

stats

add support for scipy.stats.poisson.cdf

2021-01-24 16:15:31 +00: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

Fix some doc build warnings

2021-01-25 14:08:57 -08:00

ndimage.py

[JAX] Move jax.util to jax._src_util.

2021-01-11 14:21:07 -08:00

signal.py

Fix #4775 + additional fixes

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

special.py

Add support for axis names in jnp.{sum,min,max}

2021-02-01 11:41:05 +00:00