Skip to main content

Browse the archive

swh:1:snp:71afc25eb6e6e055a37a962e6b91010ec35e397f
Revision 473d1c3685570e505bed58512afa05fb7d7a8935 authored by Skye Wanderman-Milne on 28 March 2023, 20:42:51 UTC, committed by Skye Wanderman-Milne on 28 March 2023, 22:28:13 UTC 
Turn on PJRT C API by default.
 
I forgot that the default setting is actually in jaxlib:
https://github.com/openxla/xla/blob/fbe9a80fdb8c429e8a175962459da348cd560a50/xla/python/xla_client.py#L135

To be able to make this change as a jax-only release, I manually set
the env var on Cloud TPU if it isn't already set.
1 parent 4061bbb
History
Tip revision: 0c543aef1d90f87ae20ace7a0537a30ac63d6267 authored by Yash Katariya on 17 August 2024, 00:21:10 UTC
Match the argument name with the name in `Args` section in docstring
Tip revision: 0c543ae
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-- 844 bytes
.readthedocs.yml -rw-r--r-- 569 bytes
AUTHORS -rw-r--r-- 313 bytes
CHANGELOG.md -rw-r--r-- 83.8 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.3 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
pytest.ini -rw-r--r-- 2.0 KB
setup.cfg -rw-r--r-- 1.5 KB
setup.py -rw-r--r-- 5.3 KB

README.md

back to top