[shard-map] add user tutorial

google · Jan 18, 2024 · 8b219d5 · 8b219d5
1 parent ba10775
commit 8b219d5
Show file tree

Hide file tree

Showing 4 changed files with 3,339 additions and 0 deletions.
diff --git a/docs/advanced_guide.rst b/docs/advanced_guide.rst
@@ -18,6 +18,7 @@ This section contains examples and tutorials on more advanced topics, such as Mu
 
    multi_process
    notebooks/Distributed_arrays_and_automatic_parallelization
+   notebooks/shard_map
    notebooks/xmap_tutorial
 
 .. toctree::

diff --git a/docs/conf.py b/docs/conf.py
@@ -209,6 +209,7 @@
     'notebooks/xmap_tutorial.*',
     'notebooks/Distributed_arrays_and_automatic_parallelization.*',
     'notebooks/autodiff_remat.*',
+    'notebooks/shard_map.*',
     # Requires accelerators
     'pallas/quickstart.*',
 ]