jax authors 4f8814a760 Copybara import of the project: ...

--
bf15ba5310d5f9009571928f70548bcbc7e856c3 by Matthew Johnson <mattjj@google.com>:

don't device transfer in convert_element_type

Co-authored-by: Qiao Zhang <zhangqiaorjc@google.com>
PiperOrigin-RevId: 363995032

2021-03-19 16:35:37 -07:00

..

optimize

DOC: fix update_numpydoc to prevent deleting docstring content

2021-02-22 16:45:40 -08:00

sparse

added bicgstab to new jax repo

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

stats

BUG: fix jax.scipy.stats.dirichlet implementation & add tests

2021-02-17 09:40:18 -08: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

Copybara import of the project:

2021-03-19 16:35:37 -07:00