Revision 01194bd2fb1f7b6a46db88be14324f7acdb1723e authored by Sergei Lebedev on 16 May 2024, 18:27:56 UTC, committed by jax authors on 16 May 2024, 18:29:09 UTC
The callback APIs were migrated to use jax.Arrays for both inputs and outputs in JAX 0.4.27. PiperOrigin-RevId: 634473890
1 parent 380503b
File | Mode | Size |
---|---|---|
.github | ||
benchmarks | ||
build | ||
cloud_tpu_colabs | ||
docs | ||
examples | ||
images | ||
jax | ||
jax_plugins | ||
jaxlib | ||
tests | ||
third_party | ||
.bazelrc | -rw-r--r-- | 18.4 KB |
.bazelversion | -rw-r--r-- | 6 bytes |
.editorconfig | -rw-r--r-- | 292 bytes |
.gitignore | -rw-r--r-- | 355 bytes |
.pre-commit-config.yaml | -rw-r--r-- | 1.1 KB |
.readthedocs.yml | -rw-r--r-- | 570 bytes |
AUTHORS | -rw-r--r-- | 313 bytes |
CHANGELOG.md | -rw-r--r-- | 125.3 KB |
CITATION.bib | -rw-r--r-- | 408 bytes |
CONTRIBUTING.md | -rw-r--r-- | 150 bytes |
LICENSE | -rw-r--r-- | 11.1 KB |
README.md | -rw-r--r-- | 20.0 KB |
WORKSPACE | -rw-r--r-- | 1.7 KB |
conftest.py | -rw-r--r-- | 2.5 KB |
platform_mappings | -rw-r--r-- | 371 bytes |
pyproject.toml | -rw-r--r-- | 5.9 KB |
setup.py | -rw-r--r-- | 7.2 KB |
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...