| ../ | - | |
| Common_Gotchas_in_JAX.ipynb | 61.9kB | 2026-02-14 16:43:31 |
| Common_Gotchas_in_JAX.md | 37.0kB | 2026-02-14 16:43:31 |
| Custom_derivative_rules_for_Python_code.ipynb | 115.6kB | 2026-02-14 16:43:31 |
| Custom_derivative_rules_for_Python_code.md | 36.0kB | 2026-02-14 16:43:31 |
| How_JAX_primitives_work.ipynb | 60.9kB | 2026-02-14 16:43:31 |
| How_JAX_primitives_work.md | 26.3kB | 2026-02-14 16:43:31 |
| Neural_Network_and_Data_Loading.ipynb | 19.2kB | 2026-02-14 16:43:31 |
| Neural_Network_and_Data_Loading.md | 9.2kB | 2026-02-14 16:43:31 |
| README.md | 158 B | 2026-02-14 16:43:31 |
| Writing_custom_interpreters_in_Jax.ipynb | 16.8kB | 2026-02-14 16:43:31 |
| Writing_custom_interpreters_in_Jax.md | 11.6kB | 2026-02-14 16:43:31 |
| autodiff_cookbook.ipynb | 57.7kB | 2026-02-14 16:43:31 |
| autodiff_cookbook.md | 38.4kB | 2026-02-14 16:43:31 |
| convolutions.ipynb | 526.4kB | 2026-02-14 16:43:31 |
| convolutions.md | 15.4kB | 2026-02-14 16:43:31 |
| neural_network_with_tfds_data.ipynb | 14.7kB | 2026-02-14 16:43:31 |
| neural_network_with_tfds_data.md | 9.0kB | 2026-02-14 16:43:31 |
| quickstart.ipynb | 14.1kB | 2026-02-14 16:43:31 |
| quickstart.md | 9.4kB | 2026-02-14 16:43:31 |
| thinking_in_jax.ipynb | 81.1kB | 2026-02-14 16:43:31 |
| thinking_in_jax.md | 15.7kB | 2026-02-14 16:43:31 |
| vmapped_log_probs.ipynb | 42.0kB | 2026-02-14 16:43:31 |
| vmapped_log_probs.md | 6.4kB | 2026-02-14 16:43:31 |
| xmap_tutorial.ipynb | 432.3kB | 2026-02-14 16:43:31 |
| xmap_tutorial.md | 423.7kB | 2026-02-14 16:43:31 |