Peter Hawkins 3ddcec27f2 Update minimum jaxlib version to 0.1.69.

2021-07-15 17:00:13 -04:00

..

image

prevent nans in _fill_lanczos_kernel

2021-05-20 21:49:41 -06:00

lax

Update minimum jaxlib version to 0.1.69.

2021-07-15 17:00:13 -04:00

nn

[JAX] Move contents of jax/dtypes.py to jax/_src/dtypes.py.

2021-04-07 19:35:51 -07:00

numpy

Fix rank promotion error in jnp.cov

2021-07-12 13:35:19 -07:00

ops

Add x.at[idx].get().

2021-07-07 08:51:09 -04:00

scipy

Update minimum jaxlib version to 0.1.69.

2021-07-15 17:00:13 -04: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

abstract_arrays.py

Move jax.ad_util to jax._src.ad_util.

2021-06-07 14:51:34 -07:00

ad_util.py

Move jax.ad_util to jax._src.ad_util.

2021-06-07 14:51:34 -07:00

api.py

Update minimum jaxlib version to 0.1.69.

2021-07-15 17:00:13 -04:00

cloud_tpu_init.py

Changes to make jax[tpu] work better in a docker container.

2021-07-12 17:42:46 -07:00

config.py

Make maps.EXPERIMENTAL_SPMD_LOWERING into a jax.config flag

2021-07-15 05:07:09 -07:00

custom_derivatives.py

AWN-enabled reduction over named axes in reverse-mode AD

2021-07-08 12:06:29 -07:00

dlpack.py

Update minimum jaxlib version to 0.1.69.

2021-07-15 17:00:13 -04:00

dtypes.py

Fix dtypes.issubdtype when called with "bfloat16" (as string)

2021-05-23 19:32:45 +03:00

errors.py

handle case where trace debug_info is None

2021-05-06 18:38:20 -07:00

lax_reference.py

[JAX] Make lax_reference private by moving it to jax._src.lax_reference.

2021-04-08 09:06:14 -07:00

pprint_util.py

[JAX] Move pprint_util into jax._src.

2020-11-19 06:42:19 -08:00

profiler.py

Add missing typing.Optional type annotations to function parameters.

2021-05-27 20:10:23 -07:00

random.py

Broadcast arrays manually in categorical sampling.

2021-07-07 19:56:33 +01:00

source_info_util.py

Update minimum jaxlib version to 0.1.69.

2021-07-15 17:00:13 -04:00

traceback_util.py

Update minimum jaxlib version to 0.1.69.

2021-07-15 17:00:13 -04:00

tree_util.py

unify tree_map and tree_multimap

2021-04-28 19:59:31 -07:00

util.py

Delete tuple_replace.

2021-05-19 15:29:44 -04:00