jax/docs/notebooks at main · jax-ml/jax

Name		Name	Last commit message	Last commit date
parent directory ..
Common_Gotchas_in_JAX.ipynb		Common_Gotchas_in_JAX.ipynb
Common_Gotchas_in_JAX.md		Common_Gotchas_in_JAX.md
Custom_derivative_rules_for_Python_code.ipynb		Custom_derivative_rules_for_Python_code.ipynb
Custom_derivative_rules_for_Python_code.md		Custom_derivative_rules_for_Python_code.md
Distributed_arrays_and_automatic_parallelization.ipynb		Distributed_arrays_and_automatic_parallelization.ipynb
Distributed_arrays_and_automatic_parallelization.md		Distributed_arrays_and_automatic_parallelization.md
Neural_Network_and_Data_Loading.ipynb		Neural_Network_and_Data_Loading.ipynb
Neural_Network_and_Data_Loading.md		Neural_Network_and_Data_Loading.md
README.md		README.md
Writing_custom_interpreters_in_Jax.ipynb		Writing_custom_interpreters_in_Jax.ipynb
Writing_custom_interpreters_in_Jax.md		Writing_custom_interpreters_in_Jax.md
autodiff_cookbook.ipynb		autodiff_cookbook.ipynb
autodiff_cookbook.md		autodiff_cookbook.md
autodiff_remat.ipynb		autodiff_remat.ipynb
autodiff_remat.md		autodiff_remat.md
convolutions.ipynb		convolutions.ipynb
convolutions.md		convolutions.md
neural_network_with_tfds_data.ipynb		neural_network_with_tfds_data.ipynb
neural_network_with_tfds_data.md		neural_network_with_tfds_data.md
shard_map.ipynb		shard_map.ipynb
shard_map.md		shard_map.md
thinking_in_jax.ipynb		thinking_in_jax.ipynb
thinking_in_jax.md		thinking_in_jax.md
vmapped_log_probs.ipynb		vmapped_log_probs.ipynb
vmapped_log_probs.md		vmapped_log_probs.md

README.md

For instructions on how to change and test notebooks, see Update Documentation.