https://github.com/google/jax
History
Tip revision: 7bee52e94a8f5122f54efe9a9b9927947768d40b authored by James Bradbury on 17 November 2019, 02:21:00 UTC
add tie_in to broadcast_arrays
Tip revision: 7bee52e
File Mode Size
__init__.py -rw-r--r-- 575 bytes
advi.py -rw-r--r-- 4.7 KB
datasets.py -rw-r--r-- 3.2 KB
differentially_private_sgd.py -rw-r--r-- 9.0 KB
examples_test.py -rw-r--r-- 3.5 KB
gaussian_process_regression.py -rw-r--r-- 4.6 KB
kernel_lsq.py -rw-r--r-- 2.6 KB
mnist_classifier.py -rw-r--r-- 3.2 KB
mnist_classifier_fromscratch.py -rw-r--r-- 3.1 KB
mnist_vae.py -rw-r--r-- 4.9 KB
onnx2xla.py -rw-r--r-- 4.8 KB
resnet50.py -rw-r--r-- 4.4 KB
spmd_mnist_classifier_fromscratch.py -rw-r--r-- 5.0 KB

back to top