Peter Hawkins 1b3dd65daf

Avoid lexically capturing the train_images value in MNIST VAE example. (#3947) ...

* Avoid lexically capturing the train_images value in MNIST VAE example.

This has the effect of baking in the training dataset as a constant, something that LLVM does not like that much.

* Add device_put to images.

2020-08-03 16:50:20 -04:00

..

__init__.py

move jax tests and examples to repo root

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

advi.py

Replace np -> jnp, onp -> np in examples/ (#2971)

2020-05-05 15:45:07 -04:00

control_test.py

Add ability to specify individual test targets

2020-06-29 11:08:57 -07:00

control.py

Replace np -> jnp, onp -> np in examples/ (#2971)

2020-05-05 15:45:07 -04:00

datasets.py

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

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

differentially_private_sgd.py

fix prng key reuse in differential privacy example (#3646)

2020-07-02 14:29:17 -07:00

examples_test.py

Add ability to specify individual test targets

2020-06-29 11:08:57 -07:00

gaussian_process_regression.py

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

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

kernel_lsq.py

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

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

mnist_classifier_fromscratch.py

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

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

mnist_classifier.py

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

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

mnist_vae.py

Avoid lexically capturing the train_images value in MNIST VAE example. (#3947)

2020-08-03 16:50:20 -04:00

onnx2xla.py

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

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

resnet50.py

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

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

spmd_mnist_classifier_fromscratch.py

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

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