https://github.com/google/jax
Revision 5c4e88d8799d8e24ca5bfba5a84c177a66b21f92 authored by Yash Katariya on 09 August 2023, 02:00:31 UTC, committed by jax authors on 09 August 2023, 02:55:15 UTC
1 parent 6615e23
History
Tip revision: 5c4e88d8799d8e24ca5bfba5a84c177a66b21f92 authored by Yash Katariya on 09 August 2023, 02:00:31 UTC
[Memories] Allow device_put outside jax.jit to work with numpy arrays, numpy scalars, python scalars and python ints with memory kinds
Tip revision: 5c4e88d
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-- 95.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-- 26.1 KB
WORKSPACE -rw-r--r-- 1.6 KB
conftest.py -rw-r--r-- 2.5 KB
pyproject.toml -rw-r--r-- 2.9 KB
setup.cfg -rw-r--r-- 1.5 KB
setup.py -rw-r--r-- 5.5 KB

README.md

back to top