Jake VanderPlas 12c84e7a50 Add jax.errors submodule & error troubleshooting docs

2021-03-03 12:39:12 -08:00

..

image

DOC: fix some minor formatting issues

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

lax

mark lax higher-order functions for stack trace filtering

2021-02-24 21:16:20 -08:00

nn

DOC: minor documentation & formatting fixes

2021-02-23 10:31:44 -08:00

numpy

Decouple ShardedDeviceArray from _DeviceArray

2021-02-25 13:11:32 -08:00

ops

Fix some doc build warnings

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

scipy

jax.scipy.special.logsumexp: fix b=0 corner case

2021-02-26 17:05:32 -08: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

[call_tf] Add support for DLPack to avoid copying arrays in eager mode

2021-02-13 12:49:51 +02:00

errors.py

Add jax.errors submodule & error troubleshooting docs

2021-03-03 12:39:12 -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

Merge pull request #5573 from gnecula:numpy_1_20

2021-02-05 05:29:36 -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

test source sync

2021-02-25 22:07:03 -08:00

util.py

Merge pull request #5673 from apaszke:pgather

2021-02-09 07:03:01 -08:00