https://github.com/google/jax
History
Tip revision: 849f837b6a0b317b9a4dac3d46b66d8de8a5f358 authored by jax authors on 30 September 2022, 23:25:17 UTC
Merge pull request #12609 from skye:jax2tf_import
Tip revision: 849f837
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