rocm_jax

mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00

History

The future (Python 3) has arrived; no need to request it explicitly.

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

__init__.py

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

advi.py

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

datasets.py

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

differentially_private_sgd.py

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

examples_test.py

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

gaussian_process_regression.py

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

kernel_lsq.py

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

mnist_classifier_fromscratch.py

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

mnist_classifier.py

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

mnist_vae.py

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

onnx2xla.py

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

resnet50.py

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

spmd_mnist_classifier_fromscratch.py

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