|
..
|
|
Common_Gotchas_in_JAX.ipynb
|
|
Custom_derivative_rules_for_Python_code.ipynb
|
|
Distributed_arrays_and_automatic_parallelization.ipynb
|
|
Neural_Network_and_Data_Loading.ipynb
|
|
Writing_custom_interpreters_in_Jax.ipynb
|
|
autodiff_cookbook.ipynb
|
|
autodiff_remat.ipynb
|
|
convolutions.ipynb
|
|
explicit-sharding.ipynb
|
|
neural_network_with_tfds_data.ipynb
|
|
shard_map.ipynb
|
|
thinking_in_jax.ipynb
|
|
vmapped_log_probs.ipynb
|
|
Common_Gotchas_in_JAX.md
|
|
Custom_derivative_rules_for_Python_code.md
|
|
Distributed_arrays_and_automatic_parallelization.md
|
|
Neural_Network_and_Data_Loading.md
|
|
README.md
|
|
Writing_custom_interpreters_in_Jax.md
|
|
autodiff_cookbook.md
|
|
autodiff_remat.md
|
|
convolutions.md
|
|
explicit-sharding.md
|
|
neural_network_with_tfds_data.md
|
|
shard_map.md
|
|
thinking_in_jax.md
|
|
vmapped_log_probs.md
|