https://github.com/google/jax
Revision 37f9db77f72055bb3eed2b3e5e1f99a1a2ad825b authored by Yash Katariya on 04 October 2022, 05:28:26 UTC, committed by jax authors on 04 October 2022, 05:29:03 UTC
1 parent 682e86c
History
Tip revision: 37f9db77f72055bb3eed2b3e5e1f99a1a2ad825b authored by Yash Katariya on 04 October 2022, 05:28:26 UTC
Create `Array`s from `__getitem__` and `__iter__`. This is done by `device_put`ting from the host to default device which is suboptimal. But there is a TODO to fix this!
Tip revision: 37f9db7
File Mode Size
.github
benchmarks
build
cloud_tpu_colabs
docs
examples
images
jax
jaxlib
tests
third_party
.bazelrc -rw-r--r-- 17.3 KB
.bazelversion -rw-r--r-- 6 bytes
.gitignore -rw-r--r-- 300 bytes
.pre-commit-config.yaml -rw-r--r-- 641 bytes
.readthedocs.yml -rw-r--r-- 569 bytes
AUTHORS -rw-r--r-- 313 bytes
CHANGELOG.md -rw-r--r-- 68.2 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-- 24.8 KB
WORKSPACE -rw-r--r-- 1.7 KB
conftest.py -rw-r--r-- 862 bytes
mypy.ini -rw-r--r-- 646 bytes
pylintrc -rw-r--r-- 1.6 KB
pytest.ini -rw-r--r-- 1.1 KB
setup.cfg -rw-r--r-- 486 bytes
setup.py -rw-r--r-- 4.2 KB

README.md

back to top