Roy Frostig 4c179d1b66 register the lax module for stack trace filtering

2021-02-22 21:11:41 -08:00

..

image

DOC: fix some minor formatting issues

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

lax

register the lax module for stack trace filtering

2021-02-22 21:11:41 -08:00

nn

Add support for axis names in jax.nn.initializers.variance_scaling

2021-02-04 12:38:12 +00:00

numpy

Merge pull request #4323 from tudorcebere:issue2347

2021-02-22 17:19:05 -08:00

ops

Fix some doc build warnings

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

scipy

added bicgstab to new jax repo

2021-02-18 18:01:28 -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

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

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

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

util.py

Merge pull request #5673 from apaszke:pgather

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