Peter Hawkins bc5a0b336b

Remove some uses of jax.partial. (#3131)

2020-05-18 10:19:03 -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

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

2020-05-05 15:45:07 -04: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

Remove some uses of jax.partial. (#3131)

2020-05-18 10:19:03 -04:00

examples_test.py

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

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

gaussian_process_regression.py

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

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

kernel_lsq.py

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

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

mnist_classifier_fromscratch.py

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

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

mnist_classifier.py

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

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

mnist_vae.py

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

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

onnx2xla.py

Fix ONNX mnist example (#2374)

2020-03-09 16:04:59 -04:00

resnet50.py

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

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

spmd_mnist_classifier_fromscratch.py

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

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