Toggle navigation
JUPYTER
FAQ
View on GitHub
Execute on Binder
master
branch
4008-fixes
4510-2
adagrad
all-gather-primitive
array-literals
associative-scan-docs
avals-with-names
avoid-device-interactions
beta-test-float64
cleaner-tie-in
cloud_tpu_colabs
collective-permute
cse
custom-gradient-wrapper
custom-jvp-closure-fixes-2
custom-transforms3-root
custom-transforms3
debug-nans-test-block-async
debug-nans-test-move
defvjp_notebook_fix
device-put-replicated
docfix-jit
dynamic-experiments
dynamic-scoping
einsum-optimize-default
enable-conv-test
explicit-embedding
expm-precision
fix-jax2tf-typo
fix-mypy
tag
workflow_test
test-docs
jaxlib-v0.1.56
jaxlib-v0.1.55
jaxlib-v0.1.52
jaxlib-v0.1.51
jaxlib-v0.1.50
jaxlib-v0.1.49
jaxlib-v0.1.48
jaxlib-v0.1.47
jaxlib-v0.1.46
jaxlib-v0.1.45
jaxlib-v0.1.44
jaxlib-v0.1.43
jaxlib-v0.1.42
jaxlib-v0.1.40
jaxlib-v0.1.39
jaxlib-v0.1.38
jaxlib-v0.1.37
jaxlib-v0.1.36
jaxlib-v0.1.35
jaxlib-v0.1.34
jaxlib-v0.1.33
jaxlib-v0.1.32
jax-v0.2.5
jax-v0.2.4
jax-v0.2.3
jax-v0.2.2
jax-v0.2.1
jax-v0.2.0
jax
Name
bsipocz's repositories
.github
benchmarks
build
cloud_tpu_colabs
design_notes
docs
examples
images
jax
jaxlib
tests
third_party
.gitignore
.readthedocs.yml
CHANGELOG.md
CONTRIBUTING.md
LICENSE
LICENSE_SHORT
README.md
WORKSPACE
mypy.ini
pylintrc
pytest.ini
setup.cfg
setup.py