https://github.com/google/jax
History
Tip revision: 4db046efda310e775143c15ac2ac15fa47808c31 authored by jax authors on 18 March 2024, 17:09:31 UTC
Introduce hermetic CUDA in Google ML projects.
Tip revision: 4db046e
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