Jake VanderPlas 79406757d0 Remove deprecated jax.experimental.optimizers ...

The new location is jax.example_libraries.optimizers

2022-08-09 08:50:59 -07:00

..

_src

Various debugger improvements

2022-08-08 19:54:29 -07:00

example_libraries

Merge pull request #11542 from mattjj:remove-resnet50-example

2022-07-19 16:56:56 -07:00

experimental

Remove deprecated jax.experimental.optimizers

2022-08-09 08:50:59 -07:00

image

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

interpreters

Reshape sharding spec indices to the mesh shape to preserve the old semantics.

2022-08-09 06:59:38 -07:00

lax

Add bitwise XOR reducer to lax.reduce

2022-06-15 16:56:51 +02:00

lib

Reexport jaxlib.__version as jax.lib.__version__.

2022-04-28 10:25:06 -07:00

nn

Annotate nn.initializers

2022-08-04 23:17:40 -04:00

numpy

DOC: ensure that _wraps() generates correct links to wrapped functions

2022-07-21 11:12:35 -07:00

ops

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

scipy

Implementation of jax.scipy.stats.gaussian_kde

2022-06-28 15:17:12 -04:00

tools

Refactor BUILD files to avoid individually naming Python dependencies.

2022-08-05 07:49:20 -07:00

__init__.py

Merge pull request #11653 from sharadmv:debugging-docs

2022-07-28 20:26:25 -07:00

abstract_arrays.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

ad_checkpoint.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

api_util.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

BUILD

Remove deprecated jax.experimental.optimizers

2022-08-09 08:50:59 -07:00

cloud_tpu_init.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

collect_profile.py

Add collect_profile script

2022-06-03 17:56:17 -07:00

config.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

core.py

Rolling forward #11768 after test failures caused roll-back (from use of np.empty).

2022-08-05 22:19:33 -07:00

custom_batching.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

custom_derivatives.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

custom_transpose.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

debug.py

Add debugging tools under jax.debug and documentation

2022-07-28 20:07:26 -07:00

distributed.py

Unbreak jax.distributed initialization.

2022-06-07 11:10:17 -07:00

dlpack.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

dtypes.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

errors.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

flatten_util.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

jaxpr_util.py

Fix or ignore some pytype errors.

2022-05-31 21:25:55 -07:00

linear_util.py

[dynamic-shapes] fix minor bint bugs

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

prng.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

profiler.py

jax.profiler: remove deprecated functions

2022-07-19 08:13:44 -07:00

py.typed

Improve type annotations

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

random.py

Create JAX Enhancement Proposals (JEPs).

2022-08-08 16:13:58 -04:00

stages.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

test_util.py

jax.test_util: remove deprecated test classes.

2022-06-27 11:04:50 -07:00

tree_util.py

Remove deprecated function jax.tree_util.tree_multimap

2022-07-26 09:37:27 -07:00

util.py

CI: fix flake8 ignore declarations

2022-04-21 13:44:12 -07:00

version.py

Bump version numbers after 0.3.15 release

2022-07-26 08:39:17 -07:00