Jake VanderPlas 62d9223fe8 jax.image: fix error with zero-size image resizing

2021-08-11 11:20:20 -07:00

..

image

jax.image: fix error with zero-size image resizing

2021-08-11 11:20:20 -07:00

lax

[jax2tf] Small fixes for shape polymorphism.

2021-08-10 19:51:30 +03:00

nn

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

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

numpy

Enable constructing C++ ShardedDeviceArray instead of the Python object.

2021-08-11 00:47:02 -07:00

ops

Validate shapes for boolean indices

2021-08-03 15:33:44 -07:00

scipy

DOC: many small fixes

2021-08-04 16:55:13 -07: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

Ensure zeros from AD are generated on device.

2021-07-26 20:40:13 +03:00

ad_util.py

Move jax.ad_util to jax._src.ad_util.

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

api.py

Do not unflatten trees with None values in grad.

2021-08-10 20:13:12 +00: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

Update JAX flag parsing to work when some elements of argv aren't absl friendly.

2021-08-06 07:06:17 -07:00

custom_derivatives.py

DOC: many small fixes

2021-08-04 16:55:13 -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

Fix some of the formatting and reword some of the sections.

2021-07-28 13:20:52 +01:00

lax_reference.py

Improve implementation of cbrt() in JAX.

2021-07-22 14:01:28 -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

[jax2tf] Improved testing and shape polymorphism support for lax.random.

2021-07-29 16:07:13 +03:00

source_info_util.py

Update minimum jaxlib version to 0.1.69.

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

traceback_util.py

Remove Python 3.6 compatibility code.

2021-07-29 09:09:02 -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