https://github.com/google/jax
History
Tip revision: 0118f8d568d4d94665e01f284f63f0a5d9865f5e authored by Yash Katariya on 08 December 2022, 00:01:43 UTC
Prepare for jax and jaxlib 0.4.0 release
Tip revision: 0118f8d
File Mode Size
Common_Gotchas_in_JAX.ipynb -rw-r--r-- 60.4 KB
Common_Gotchas_in_JAX.md -rw-r--r-- 36.1 KB
Custom_derivative_rules_for_Python_code.ipynb -rw-r--r-- 112.9 KB
Custom_derivative_rules_for_Python_code.md -rw-r--r-- 35.2 KB
How_JAX_primitives_work.ipynb -rw-r--r-- 59.4 KB
How_JAX_primitives_work.md -rw-r--r-- 25.7 KB
Neural_Network_and_Data_Loading.ipynb -rw-r--r-- 18.8 KB
Neural_Network_and_Data_Loading.md -rw-r--r-- 9.0 KB
Parallelism_with_JAX.ipynb -rw-r--r-- 70.5 KB
Parallelism_with_JAX.md -rw-r--r-- 27.0 KB
README.md -rw-r--r-- 158 bytes
Writing_custom_interpreters_in_Jax.ipynb -rw-r--r-- 16.4 KB
Writing_custom_interpreters_in_Jax.md -rw-r--r-- 11.4 KB
autodiff_cookbook.ipynb -rw-r--r-- 56.4 KB
autodiff_cookbook.md -rw-r--r-- 37.5 KB
convolutions.ipynb -rw-r--r-- 514.1 KB
convolutions.md -rw-r--r-- 15.1 KB
neural_network_with_tfds_data.ipynb -rw-r--r-- 14.3 KB
neural_network_with_tfds_data.md -rw-r--r-- 8.8 KB
quickstart.ipynb -rw-r--r-- 16.4 KB
quickstart.md -rw-r--r-- 9.8 KB
thinking_in_jax.ipynb -rw-r--r-- 79.2 KB
thinking_in_jax.md -rw-r--r-- 15.4 KB
vmapped_log_probs.ipynb -rw-r--r-- 41.0 KB
vmapped_log_probs.md -rw-r--r-- 6.3 KB
xmap_tutorial.ipynb -rw-r--r-- 422.1 KB
xmap_tutorial.md -rw-r--r-- 413.7 KB

README.md

back to top