|
..
|
|
Common_Gotchas_in_JAX.ipynb
|
|
Custom_derivative_rules_for_Python_code.ipynb
|
|
Distributed_arrays_and_automatic_parallelization.ipynb
|
|
How_JAX_primitives_work.ipynb
|
|
Neural_Network_and_Data_Loading.ipynb
|
|
Writing_custom_interpreters_in_Jax.ipynb
|
|
autodiff_cookbook.ipynb
|
|
convolutions.ipynb
|
|
external_callbacks.ipynb
|
|
neural_network_with_tfds_data.ipynb
|
|
quickstart.ipynb
|
|
thinking_in_jax.ipynb
|
|
vmapped_log_probs.ipynb
|
|
xmap_tutorial.ipynb
|
|
Common_Gotchas_in_JAX.md
|
|
Custom_derivative_rules_for_Python_code.md
|
|
Distributed_arrays_and_automatic_parallelization.md
|
|
How_JAX_primitives_work.md
|
|
Neural_Network_and_Data_Loading.md
|
|
README.md
|
|
Writing_custom_interpreters_in_Jax.md
|
|
autodiff_cookbook.md
|
|
convolutions.md
|
|
external_callbacks.md
|
|
neural_network_with_tfds_data.md
|
|
quickstart.md
|
|
thinking_in_jax.md
|
|
vmapped_log_probs.md
|
|
xmap_tutorial.md
|