https://github.com/google/jax
History
Tip revision: 9e4114f0f13fec49e054240b344dad9e56f988dc authored by Yash Katariya on 27 September 2022, 17:06:10 UTC
Move `array.py` and `sharding.py` from `experimental/` to `_src/`.
Tip revision: 9e4114f
File Mode Size
10657-sequencing-effects.md -rw-r--r-- 13.1 KB
11830-new-remat-checkpoint.md -rw-r--r-- 6.5 KB
12049-type-annotations.md -rw-r--r-- 29.7 KB
2026-custom-derivatives.md -rw-r--r-- 19.1 KB
263-prng.md -rw-r--r-- 11.3 KB
4008-custom-vjp-update.md -rw-r--r-- 4.9 KB
4410-omnistaging.md -rw-r--r-- 13.8 KB
9407-type-promotion.ipynb -rw-r--r-- 395.5 KB
9407-type-promotion.md -rw-r--r-- 42.4 KB
9419-jax-versioning.md -rw-r--r-- 9.2 KB
index.rst -rw-r--r-- 2.2 KB

back to top