https://github.com/google/jax
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
lorentz.png -rw-r--r-- 560.6 KB
nested_pmap.png -rw-r--r-- 23.3 KB
wave_movie.gif -rw-r--r-- 5.6 MB

back to top