jax authors b6318f89c4 Merge pull request #7993 from gnecula:tf_share_const ...

PiperOrigin-RevId: 398504028

2021-09-23 09:23:06 -07:00

..

_src

Merge pull request #7983 from jakevdp:fix-std-basis

2021-09-23 09:15:51 -07:00

experimental

Merge pull request #7993 from gnecula:tf_share_const

2021-09-23 09:23:06 -07:00

image

Add PEP484-compatible export for jax and its subpackages

2021-09-13 14:08:48 -07:00

interpreters

Move contents of jax.lib to jax._src.lib.

2021-09-23 06:33:55 -07:00

lax

Remove export of jax.lax.partial.

2021-09-14 16:17:50 -04:00

lib

Move contents of jax.lib to jax._src.lib.

2021-09-23 06:33:55 -07:00

nn

Fix package exports

2021-09-14 13:55:55 -07:00

numpy

jax.numpy: add wrapper for np.insert

2021-09-16 12:18:32 -07:00

ops

Add PEP484-compatible export for jax and its subpackages

2021-09-13 14:08:48 -07:00

scipy

Fix package exports

2021-09-14 13:55:55 -07:00

tools

Move contents of jax.lib to jax._src.lib.

2021-09-23 06:33:55 -07:00

__init__.py

Move contents of jax.lib to jax._src.lib.

2021-09-23 06:33:55 -07:00

abstract_arrays.py

Move jax.abstract_arrays to jax._src.abstract_arrays.

2021-06-02 06:25:22 -07:00

api_util.py

Move contents of jax.api_util to jax._src.api_util and add a forwarding shim.

2021-09-08 09:00:56 -07:00

cloud_tpu_init.py

Automatically initialize Cloud TPU topology env vars if running on a Cloud TPU VM.

2021-03-10 09:15:31 -08:00

config.py

Move jax.config to jax._src.config.

2021-04-19 08:53:12 -07:00

core.py

Switch internal users of jax.util.partial to use functools.partial.

2021-09-13 21:09:58 -04:00

custom_derivatives.py

Add PEP484-compatible export for jax and its subpackages

2021-09-13 14:08:48 -07:00

dlpack.py

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

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

dtypes.py

Add PEP484-compatible export for jax and its subpackages

2021-09-13 14:08:48 -07:00

errors.py

Add PEP484-compatible export for jax and its subpackages

2021-09-13 14:08:48 -07:00

flatten_util.py

Move implementation of jax.flatten_util to jax._src.flatten_util. Add a jax.flatten_util shim.

2021-09-08 13:54:25 -07:00

jaxpr_util.py

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

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

linear_util.py

Cleanup: switch to new version of super()

2021-08-05 13:11:07 -07:00

prng.py

Add PEP484-compatible export for jax and its subpackages

2021-09-13 14:08:48 -07:00

profiler.py

Add PEP484-compatible export for jax and its subpackages

2021-09-13 14:08:48 -07:00

py.typed

Improve type annotations

2021-01-13 10:26:35 -05:00

random.py

Merge pull request #7847 from google:prng-type

2021-09-14 12:55:28 -07:00

test_util.py

Move contents of jax.lib to jax._src.lib.

2021-09-23 06:33:55 -07:00

tree_util.py

Add PEP484-compatible export for jax and its subpackages

2021-09-13 14:08:48 -07:00

util.py

Remove jax.util.partial.

2021-09-20 20:32:49 -04:00

version.py

update version and changelog for pypi

2021-09-02 15:38:47 -07:00