https://github.com/google/jax
History
Tip revision: 7e1a3799e49a83d89d6501ca4507760b21141bbe authored by Parker Schuh on 10 March 2023, 02:08:09 UTC
_arrays returns PyArray instead of PyBuffer.
Tip revision: 7e1a379
File Mode Size
01-jax-basics.ipynb -rw-r--r-- 43.9 KB
01-jax-basics.md -rw-r--r-- 14.6 KB
02-jitting.ipynb -rw-r--r-- 57.8 KB
02-jitting.md -rw-r--r-- 13.6 KB
03-vectorization.ipynb -rw-r--r-- 9.0 KB
03-vectorization.md -rw-r--r-- 4.7 KB
04-advanced-autodiff.ipynb -rw-r--r-- 20.5 KB
04-advanced-autodiff.md -rw-r--r-- 12.8 KB
05-random-numbers.ipynb -rw-r--r-- 15.8 KB
05-random-numbers.md -rw-r--r-- 9.2 KB
05.1-pytrees.ipynb -rw-r--r-- 34.5 KB
05.1-pytrees.md -rw-r--r-- 10.9 KB
06-parallelism.ipynb -rw-r--r-- 44.7 KB
06-parallelism.md -rw-r--r-- 16.1 KB
07-state.ipynb -rw-r--r-- 28.6 KB
07-state.md -rw-r--r-- 9.2 KB
08-pjit.rst -rw-r--r-- 370 bytes
index.rst -rw-r--r-- 355 bytes

back to top