https://github.com/google/jax
History
Tip revision: c75a532223772fee2353adf81f9f1bbf2cacdbd2 authored by Peter Hawkins on 14 February 2024, 02:34:28 UTC
Micro-optimization: Avoid allocating a redundant SourceInfo in default_process_primitive.
Tip revision: c75a532
File Mode Size
01-jax-basics.ipynb -rw-r--r-- 43.8 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.7 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-- 21.5 KB
04-advanced-autodiff.md -rw-r--r-- 13.4 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-- 53.2 KB
05.1-pytrees.md -rw-r--r-- 16.8 KB
06-parallelism.ipynb -rw-r--r-- 43.6 KB
06-parallelism.md -rw-r--r-- 15.7 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-- 368 bytes
index.rst -rw-r--r-- 355 bytes

back to top