https://github.com/google/jax
Revision ef09973a8855097ba842c6d6920d2fac136772d6 authored by Sergei Lebedev on 30 April 2024, 19:41:41 UTC, committed by jax authors on 01 May 2024, 12:22:52 UTC
The new flag passes DLPack tensors for both arguments and results into the
callback, allowing to do zero-copy interop with other tensor programming
libraries.

See #20701 for motivation.

PiperOrigin-RevId: 629497642
1 parent cbee909
History
Tip revision: ef09973a8855097ba842c6d6920d2fac136772d6 authored by Sergei Lebedev on 30 April 2024, 19:41:41 UTC
dlpack.callback now supports mutable_results=
Tip revision: ef09973
File Mode Size
.github
benchmarks
build
cloud_tpu_colabs
docs
examples
images
jax
jax_plugins
jaxlib
tests
third_party
.bazelrc -rw-r--r-- 18.1 KB
.bazelversion -rw-r--r-- 6 bytes
.editorconfig -rw-r--r-- 292 bytes
.gitignore -rw-r--r-- 355 bytes
.pre-commit-config.yaml -rw-r--r-- 1.1 KB
.readthedocs.yml -rw-r--r-- 570 bytes
AUTHORS -rw-r--r-- 313 bytes
CHANGELOG.md -rw-r--r-- 123.3 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-- 20.0 KB
WORKSPACE -rw-r--r-- 568 bytes
conftest.py -rw-r--r-- 2.5 KB
platform_mappings -rw-r--r-- 371 bytes
pyproject.toml -rw-r--r-- 5.9 KB
setup.py -rw-r--r-- 7.2 KB

README.md

back to top