Toggle navigation
JUPYTER
FAQ
View on GitHub
Execute on Binder
remove-prng-maxsize-restriction
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
docs
Name
..
README.md
conf.py
index.rst
jax.experimental.optimizers.rst
jax.experimental.rst
jax.experimental.stax.rst
jax.lax.rst
jax.numpy.rst
jax.ops.rst
jax.random.rst
jax.rst
jax.scipy.rst
modules.rst
requirements.txt