https://github.com/google/jax
Revision 9d1cbc7d21d8ca0fbe3de9bc16898d7df383893a authored by Skye Wanderman-Milne on 22 August 2023, 20:10:12 UTC, committed by Skye Wanderman-Milne on 23 August 2023, 00:06:16 UTC
I messed up the forwards compat in
https://github.com/google/jax/commit/3e50fea29edb6b78426dde511414429f2d2fddf8. The
next jaxlib release won't need the env var at all, but jaxlib 0.4.14
and older still do.
1 parent 27d19ee
History
Tip revision: 9d1cbc7d21d8ca0fbe3de9bc16898d7df383893a authored by Skye Wanderman-Milne on 22 August 2023, 20:10:12 UTC
Default to PJRT TPU runtime instead of StreamExecutor on older jaxlibs.
Tip revision: 9d1cbc7
File Mode Size
.github
benchmarks
build
cloud_tpu_colabs
docs
examples
images
jax
jaxlib
tests
third_party
.bazelrc -rw-r--r-- 14.0 KB
.bazelversion -rw-r--r-- 6 bytes
.editorconfig -rw-r--r-- 292 bytes
.gitignore -rw-r--r-- 342 bytes
.pre-commit-config.yaml -rw-r--r-- 1.0 KB
.readthedocs.yml -rw-r--r-- 569 bytes
AUTHORS -rw-r--r-- 313 bytes
CHANGELOG.md -rw-r--r-- 97.0 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.9 KB
WORKSPACE -rw-r--r-- 633 bytes
conftest.py -rw-r--r-- 2.5 KB
pyproject.toml -rw-r--r-- 3.0 KB
setup.cfg -rw-r--r-- 1.5 KB
setup.py -rw-r--r-- 5.6 KB

README.md

back to top