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 |