jax authors b5e6145a42 Merge pull request #11359 from hawkinsp:bazel ...

PiperOrigin-RevId: 459234031

2022-07-06 06:13:20 -07:00

..

jax_cpp

Merge pull request #11359 from hawkinsp:bazel

2022-07-06 06:13:20 -07:00

__init__.py

move jax tests and examples to repo root

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

advi.py

feat: refactor code using pyupgrade

2022-05-17 22:14:05 +01:00

control_test.py

[x64] make examples/control_test compatible with strict dtype promotion

2022-06-16 16:20:54 -07:00

control.py

[x64] make examples/control_test compatible with strict dtype promotion

2022-06-16 16:20:54 -07:00

datasets.py

feat: refactor code using pyupgrade

2022-05-17 22:14:05 +01:00

differentially_private_sgd.py

Migrate from deprecated tensorflow_privacy RDP accountant to differential_privacy.

2022-06-13 16:26:51 -07:00

examples_test.py

feat: refactor code using pyupgrade

2022-05-17 22:14:05 +01:00

gaussian_process_regression.py

feat: refactor code using pyupgrade

2022-05-17 22:14:05 +01: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

feat: refactor code using pyupgrade

2022-05-17 22:14:05 +01:00

mnist_classifier.py

feat: refactor code using pyupgrade

2022-05-17 22:14:05 +01:00

mnist_vae.py

feat: refactor code using pyupgrade

2022-05-17 22:14:05 +01: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

feat: refactor code using pyupgrade

2022-05-17 22:14:05 +01:00