Roy Frostig 623c201054 [JAX] move example libraries from jax.experimental into jax.example_libraries ...

The `jax.experimental.stax` and `jax.experimental.optimizers` modules are standalone examples libraries. By contrast, the remaining modules in `jax.experimental` are experimental features of the JAX core system. This change moves the two example libraries, and the README that describes them, to `jax.example_libraries` to reflect this distinction.

PiperOrigin-RevId: 404405186

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

..

jax_cpp

Remove redundant dep in jax cpp example target.

2021-06-22 14:49:19 -07:00

__init__.py

move jax tests and examples to repo root

2018-11-18 15:15:47 -08:00

advi.py

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

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

control_test.py

Move jax.test_util to jax._src.test_util.

2021-09-24 07:02:49 -07:00

control.py

[JAX] Replace uses of deprecated jax.ops.index_update(x, idx, y) APIs with their up-to-date, more succinct equivalent x.at[idx].set(y).

2021-10-01 08:45:31 -07:00

datasets.py

Remove "from __future__" uses from JAX. (#2117)

2020-01-29 12:29:03 -05:00

differentially_private_sgd.py

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

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

examples_test.py

Move jax.test_util to jax._src.test_util.

2021-09-24 07:02:49 -07:00

gaussian_process_regression.py

[JAX] Fix pylint errors.

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

kernel_lsq.py

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

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

mnist_classifier_fromscratch.py

Revert import path changes to examples/ and benchmarks/

2021-01-20 17:35:55 -08:00

mnist_classifier.py

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

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

mnist_vae.py

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

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

onnx2xla.py

Fix remaining flakes and use exclude within setup.cfg (#3371)

2020-06-08 22:58:03 -07:00

resnet50.py

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

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

spmd_mnist_classifier_fromscratch.py

Move contents of jax.lib to jax._src.lib.

2021-09-23 06:33:55 -07:00