https://github.com/google/jax
Revision 0640df71de9f5e8eefafab36623ab0789fcc0e4d authored by Colin Gaffney on 05 April 2023, 23:09:45 UTC, committed by jax authors on 06 April 2023, 16:22:22 UTC
1 parent b8ade58
History
Tip revision: 0640df71de9f5e8eefafab36623ab0789fcc0e4d authored by Colin Gaffney on 05 April 2023, 23:09:45 UTC
Fix slow repeated call to check whether a checkpoint is in OCDBT format or not. Enable options to speed up OCDBT serialize/deserialize.
Tip revision: 0640df7
File Mode Size
.github
benchmarks
build
cloud_tpu_colabs
docs
examples
images
jax
jaxlib
tests
third_party
.bazelrc -rw-r--r-- 14.7 KB
.bazelversion -rw-r--r-- 6 bytes
.gitignore -rw-r--r-- 342 bytes
.pre-commit-config.yaml -rw-r--r-- 993 bytes
.readthedocs.yml -rw-r--r-- 569 bytes
AUTHORS -rw-r--r-- 313 bytes
CHANGELOG.md -rw-r--r-- 84.4 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-- 25.7 KB
WORKSPACE -rw-r--r-- 1.6 KB
conftest.py -rw-r--r-- 2.5 KB
mypy.ini -rw-r--r-- 1.3 KB
pylintrc -rw-r--r-- 1.6 KB
pyproject.toml -rw-r--r-- 90 bytes
pytest.ini -rw-r--r-- 2.0 KB
setup.cfg -rw-r--r-- 1.5 KB
setup.py -rw-r--r-- 5.4 KB

README.md

back to top