Skip to main content

Browse the archive

File Mode Size
Common_Gotchas_in_JAX.ipynb -rw-r--r-- 67.0 KB
Common_Gotchas_in_JAX.md -rw-r--r-- 39.4 KB
Custom_derivative_rules_for_Python_code.ipynb -rw-r--r-- 113.1 KB
Custom_derivative_rules_for_Python_code.md -rw-r--r-- 35.4 KB
Distributed_arrays_and_automatic_parallelization.ipynb -rw-r--r-- 71.1 KB
Distributed_arrays_and_automatic_parallelization.md -rw-r--r-- 27.5 KB
How_JAX_primitives_work.ipynb -rw-r--r-- 59.7 KB
How_JAX_primitives_work.md -rw-r--r-- 26.0 KB
Neural_Network_and_Data_Loading.ipynb -rw-r--r-- 19.0 KB
Neural_Network_and_Data_Loading.md -rw-r--r-- 9.2 KB
README.md -rw-r--r-- 158 bytes
Writing_custom_interpreters_in_Jax.ipynb -rw-r--r-- 16.6 KB
Writing_custom_interpreters_in_Jax.md -rw-r--r-- 11.6 KB
autodiff_cookbook.ipynb -rw-r--r-- 56.6 KB
autodiff_cookbook.md -rw-r--r-- 37.7 KB
autodiff_remat.ipynb -rw-r--r-- 91.1 KB
autodiff_remat.md -rw-r--r-- 20.1 KB
convolutions.ipynb -rw-r--r-- 514.2 KB
convolutions.md -rw-r--r-- 15.3 KB
external_callbacks.ipynb -rw-r--r-- 29.2 KB
external_callbacks.md -rw-r--r-- 5.5 KB
neural_network_with_tfds_data.ipynb -rw-r--r-- 14.5 KB
neural_network_with_tfds_data.md -rw-r--r-- 9.0 KB
quickstart.ipynb -rw-r--r-- 16.6 KB
quickstart.md -rw-r--r-- 9.9 KB
thinking_in_jax.ipynb -rw-r--r-- 79.4 KB
thinking_in_jax.md -rw-r--r-- 15.5 KB
vmapped_log_probs.ipynb -rw-r--r-- 41.2 KB
vmapped_log_probs.md -rw-r--r-- 6.5 KB
xmap_tutorial.ipynb -rw-r--r-- 422.1 KB
xmap_tutorial.md -rw-r--r-- 413.7 KB

README.md

back to top