Index of /open-courses-0923/02-CS224U-Natural-Language-Understanding/github/repair-google--jax/docs/notebooks/

../
Common_Gotchas_in_JAX.ipynb                        14-Oct-2026 11:08     50K
Common_Gotchas_in_JAX.md                           14-Oct-2026 11:08     27K
Custom_derivative_rules_for_Python_code.ipynb      14-Oct-2026 11:08    114K
Custom_derivative_rules_for_Python_code.md         14-Oct-2026 11:08     36K
Neural_Network_and_Data_Loading.ipynb              14-Oct-2026 11:08     20K
Neural_Network_and_Data_Loading.md                 14-Oct-2026 11:08    9026
README.md                                          14-Oct-2026 11:08     152
Writing_custom_interpreters_in_Jax.ipynb           14-Oct-2026 11:08     17K
Writing_custom_interpreters_in_Jax.md              14-Oct-2026 11:08     12K
autodiff_cookbook.ipynb                            14-Oct-2026 11:08     57K
autodiff_cookbook.md                               14-Oct-2026 11:08     38K
autodiff_remat.ipynb                               14-Oct-2026 11:08     90K
autodiff_remat.md                                  14-Oct-2026 11:08     20K
colocated-python.ipynb                             14-Oct-2026 11:08     13K
colocated-python.md                                14-Oct-2026 11:08    9425
cute_dsl_jax.ipynb                                 14-Oct-2026 11:08     66K
cute_dsl_jax.md                                    14-Oct-2026 11:08     50K
layout.ipynb                                       14-Oct-2026 11:08    9927
layout.md                                          14-Oct-2026 11:08    6806
neural_network_with_tfds_data.ipynb                14-Oct-2026 11:08     15K
neural_network_with_tfds_data.md                   14-Oct-2026 11:08    9300
shard_map.ipynb                                    14-Oct-2026 11:08      2M
shard_map.md                                       14-Oct-2026 11:08      2M
shard_map.py                                       14-Oct-2026 11:08      2M
thinking_in_jax.ipynb                              14-Oct-2026 11:08     29K
thinking_in_jax.md                                 14-Oct-2026 11:08     20K
vmapped_log_probs.ipynb                            14-Oct-2026 11:08     41K
vmapped_log_probs.md                               14-Oct-2026 11:08    6635