Jake VanderPlas 5e7be4a61f Cleanup: remove obsolete jaxlib version checks

2021-02-04 15:13:39 -08:00

..

image

DOC: fix some minor formatting issues

2021-01-28 15:20:02 -08:00

lax

Tightened the API of some jax.numpy functions

2021-02-03 11:06:18 +02:00

nn

Add support for named axes in jax.nn.one_hot

2021-02-02 15:57:00 +00:00

numpy

Fix a _check_arraylike error.

2021-02-04 13:55:36 +00:00

ops

Fix some doc build warnings

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

scipy

Add logdf and pdf for chisquare distribution

2021-02-03 15:09:21 -05:00

third_party

Move jax.third_party to jax._src.third_party.

2020-10-17 11:43:07 -07:00

__init__.py

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

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

dlpack.py

Cleanup: remove obsolete jaxlib version checks

2021-02-04 15:13:39 -08:00

pprint_util.py

[JAX] Move pprint_util into jax._src.

2020-11-19 06:42:19 -08:00

profiler.py

[JAX] Move implementation of profiler.py into jax._src.

2020-11-04 09:06:40 -08:00

random.py

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

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

source_info_util.py

Revert part of #5319: don't use traceback_util.path_starts in source_info_util.user_frame.

2021-01-19 15:01:52 -08:00

traceback_util.py

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

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

util.py

Make jax_enable_x64 a thread-local value.

2021-02-04 09:48:22 -08:00