Toggle navigation
JUPYTER
FAQ
View on GitHub
Execute on Binder
parallel
branch
adagrad
automasking
beta-test-float64
collective-permute
cse
custom-transforms
fix-notebooks
fix-notebooks2
fix-repr
fix-travis
issue744
issue768
issue831
jb/cond-batching-rule
jb/initializers
jet
master
no-more-tuples
papply
parallel
pmap
pmap-docs
primitives
prng-slow
prngkey-linearity-brainstorming
ray
remove-prng-maxsize-restriction
shoyer-jit-stuff
software-pipeline
tuple-simplification
tag
jax
examples
Name
..
__init__.py
advi.py
datasets.py
examples_test.py
kernel_lsq.py
mnist_classifier.py
mnist_classifier_fromscratch.py
mnist_vae.py
onnx2xla.py
resnet50.py