https://github.com/google/jax
History
Tip revision: 3a0214706e3d96849f7a2a535937bf7cdfa075dc authored by Bixia Zheng on 15 August 2022, 22:20:02 UTC
Change jax to lower bessel_i1e primitive to chlo.bessel_i1e.
Tip revision: 3a02147
File Mode Size
01-jax-basics.ipynb -rw-r--r-- 43.6 KB
01-jax-basics.md -rw-r--r-- 14.4 KB
02-jitting.ipynb -rw-r--r-- 57.6 KB
02-jitting.md -rw-r--r-- 13.5 KB
03-vectorization.ipynb -rw-r--r-- 8.8 KB
03-vectorization.md -rw-r--r-- 4.5 KB
04-advanced-autodiff.ipynb -rw-r--r-- 20.3 KB
04-advanced-autodiff.md -rw-r--r-- 12.6 KB
05-random-numbers.ipynb -rw-r--r-- 15.7 KB
05-random-numbers.md -rw-r--r-- 9.0 KB
05.1-pytrees.ipynb -rw-r--r-- 34.2 KB
05.1-pytrees.md -rw-r--r-- 10.7 KB
06-parallelism.ipynb -rw-r--r-- 44.5 KB
06-parallelism.md -rw-r--r-- 15.9 KB
07-state.ipynb -rw-r--r-- 28.5 KB
07-state.md -rw-r--r-- 9.0 KB
08-pjit.md -rw-r--r-- 19.7 KB
index.rst -rw-r--r-- 341 bytes

back to top