Peter Hawkins 4dc69034b0 Update version numbers after jax/jaxlib release.

2022-04-07 16:40:19 -04:00

..

_src

Minor change to lax to support jax2tf shape polymorphic concatenation.

2022-04-07 08:34:27 -07:00

example_libraries

Rename jax.nn.normalize to standardize. Add normalize alias with DeprecationWarning.

2022-03-23 20:55:22 +00:00

experimental

Merge pull request #10184 from jakevdp:merge-bcoo-dot-general

2022-04-07 12:57:19 -07:00

image

Documentation improvements.

2022-03-08 09:37:33 -05:00

interpreters

Move PartitionSpec from sharded_jit.py to pxla.py. The public endpoint is via jax.experimental so that should be used (no changes to the public endpoint).

2022-04-06 15:19:19 -07:00

lax

remove _select_and_{gather,scatter}_add from public jax.lax module

2022-03-10 10:43:42 -08:00

lib

Use imports relative to the jax package consistently, rather than .-relative imports.

2021-11-24 07:48:29 -08:00

nn

Rename jax.nn.normalize to standardize. Add normalize alias with DeprecationWarning.

2022-03-23 20:55:22 +00:00

numpy

Better error messages for jnp.fromiter and jnp.fromfile

2022-03-29 14:30:32 -07:00

ops

Remove jax.ops.index... functions.

2022-02-24 09:36:28 -05:00

scipy

Add jax.scipy.linalg.rsf2csf

2022-04-06 21:06:23 +01:00

tools

Fix SyntaxError in jax_to_ir

2021-12-21 15:11:08 -08:00

__init__.py

Add version int tuple __version_info__ to JAX

2022-04-05 13:26:05 -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

…

config.py

…

core.py

remove string annotations from core.py

2022-04-03 11:19:07 -07:00

custom_batching.py

introduce custom_batching.sequential_vmap

2022-01-20 21:37:58 -08:00

custom_derivatives.py

Add PEP484-compatible export for jax and its subpackages

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

custom_transpose.py

introduce custom_transpose and a corresponding primitive

2022-01-11 12:51:17 -08:00

distributed.py

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

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

dlpack.py

…

dtypes.py

Add PEP484-compatible export for jax and its subpackages

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

errors.py

Use imports relative to the jax package consistently, rather than .-relative imports.

2021-11-24 07:48:29 -08: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

Increase minimum jaxlib version to 0.3.0.

2022-03-04 10:33:03 -05:00

linear_util.py

clean up WrappedFun.call_wrapped refs on exception

2022-01-05 16:11:48 -08: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

…

random.py

add random.loggamma and improve dirichlet & beta implementation

2022-03-21 08:33:11 -07:00

stages.py

introduce a protocol for compilation wrappers

2022-03-17 17:13:33 -07:00

test_util.py

Separate jax.test_util implementations into public and private sources.

2022-04-04 14:43:39 -07:00

tree_util.py

Deprecate jax.tree_util.tree_multimap

2022-04-01 14:51:54 -07:00

util.py

Remove jax.util.partial.

2021-09-28 11:04:02 -04:00

version.py

Update version numbers after jax/jaxlib release.

2022-04-07 16:40:19 -04:00