https://github.com/google/jax
History
Tip revision: bfe8f8732b9b90a3a0e7d79c40964b759833bdea authored by George Necula on 02 July 2020, 09:47:22 UTC
Change the version of tensorflow, the one in jaxlib 0.1.51 does not build (#3637)
Tip revision: bfe8f87
File Mode Size
__init__.py -rw-r--r-- 575 bytes
advi.py -rw-r--r-- 4.8 KB
control.py -rw-r--r-- 5.9 KB
control_test.py -rw-r--r-- 7.7 KB
datasets.py -rw-r--r-- 3.1 KB
differentially_private_sgd.py -rw-r--r-- 8.9 KB
examples_test.py -rw-r--r-- 3.3 KB
gaussian_process_regression.py -rw-r--r-- 4.5 KB
kernel_lsq.py -rw-r--r-- 2.5 KB
mnist_classifier.py -rw-r--r-- 3.1 KB
mnist_classifier_fromscratch.py -rw-r--r-- 3.0 KB
mnist_vae.py -rw-r--r-- 4.8 KB
onnx2xla.py -rw-r--r-- 4.7 KB
resnet50.py -rw-r--r-- 4.3 KB
spmd_mnist_classifier_fromscratch.py -rw-r--r-- 4.9 KB

back to top