Add thinking_in_jax.ipynb

jax-ml · Jan 26, 2021 · cba5284 · cba5284
1 parent 84f55f8
commit cba5284
Show file tree

Hide file tree

Showing 2 changed files with 1,252 additions and 0 deletions.
diff --git a/docs/index.rst b/docs/index.rst
@@ -12,6 +12,7 @@ For an introduction to JAX, start at the
    :caption: Tutorials
 
    notebooks/quickstart
+   thinking_in_jax
    notebooks/autodiff_cookbook
    notebooks/vmapped_log_probs
    Training a Simple Neural Network, with Tensorflow Datasets Data Loading <https://github.com/google/jax/blob/master/docs/notebooks/neural_network_with_tfds_data.ipynb>