Qiao Zhang ad4cb94734 update version and changelog for pypi

2021-11-10 14:21:26 -08:00

..

_src

Merge pull request #8498 from jakevdp:meshgrid-args

2021-11-09 23:44:41 -08:00

example_libraries

[JAX] move example libraries from jax.experimental into jax.example_libraries

2021-10-19 17:30:45 -07:00

experimental

Merge pull request #8457 from dan-zheng:fix-typos

2021-11-10 04:24:45 -08:00

image

Add PEP484-compatible export for jax and its subpackages

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

interpreters

Merge pull request #8451 from mattjj:wip-merging-typeclasses

2021-11-04 16:04:07 -07:00

lax

[JAX] Added jax.lax.linalg.qdwh.

2021-10-29 14:45:06 -07:00

lib

Remove xla_bridge.{constant, register_constant_handler, _python_scalar_constant} from API.

2021-10-18 13:56:58 -07:00

nn

Fix package exports

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

numpy

jax.numpy: add missing uint definition

2021-11-01 10:05:27 -07:00

ops

Add PEP484-compatible export for jax and its subpackages

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

scipy

[JAX] Fix pylint errors.

2021-10-04 17:54:46 -07:00

tools

colab_tpu.setup_tpu: allow user specification ot tpu driver version

2021-11-08 08:59:16 -08:00

__init__.py

Add jax.distributed.initialize for multi-host GPU.

2021-10-26 14:37:54 -07:00

abstract_arrays.py

Move jax.abstract_arrays to jax._src.abstract_arrays.

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

ad_checkpoint.py

checkpoint_name for checkpoint policies by name

2021-10-13 21:31:02 -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

jax_check_tracer_leaks: add warning about debuggers

2021-11-08 09:21:18 -08:00

custom_derivatives.py

Add PEP484-compatible export for jax and its subpackages

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

distributed.py

Add jax.distributed.initialize for multi-host GPU.

2021-10-26 14:37:54 -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

Change jax.core.DropVar to be a non-singleton.

2021-10-18 15:02:54 -07:00

linear_util.py

[JAX] Fix pylint errors.

2021-10-04 17:54:46 -07:00

prng.py

improvements to RBG PRNG

2021-10-07 18:59:13 -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

add a config setting to control the default PRNG implementation

2021-10-07 21:22:40 -07:00

test_util.py

Readd jax.test_util.check_jvp and check_vjp to the public JAX API.

2021-10-14 11:55:11 -04: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-28 11:04:02 -04:00

version.py

update version and changelog for pypi

2021-11-10 14:21:26 -08:00