| File | Mode | Size |
|---|---|---|
| Common_Gotchas_in_JAX.ipynb | -rw-r--r-- | 378.7 KB |
| Custom_derivative_rules_for_Python_code.ipynb | -rw-r--r-- | 131.2 KB |
| How_JAX_primitives_work.ipynb | -rw-r--r-- | 68.6 KB |
| Neural_Network_and_Data_Loading.ipynb | -rw-r--r-- | 19.7 KB |
| README.md | -rw-r--r-- | 158 bytes |
| Writing_custom_interpreters_in_Jax.ipynb | -rw-r--r-- | 19.7 KB |
| XLA_in_Python.ipynb | -rw-r--r-- | 47.2 KB |
| autodiff_cookbook.ipynb | -rw-r--r-- | 66.7 KB |
| maml.ipynb | -rw-r--r-- | 209.1 KB |
| neural_network_with_tfds_data.ipynb | -rw-r--r-- | 14.7 KB |
| quickstart.ipynb | -rw-r--r-- | 16.2 KB |
| score_matching.ipynb | -rw-r--r-- | 2.4 MB |
| vmapped_log_probs.ipynb | -rw-r--r-- | 44.8 KB |
