Merge pull request #346 from fllinares:custom_loop_pmap_example

PiperOrigin-RevId: 489975655
google · Nov 21, 2022 · 238fcb4 · 238fcb4
2 parents 3828913 + 453dc87
commit 238fcb4
Show file tree

Hide file tree

Showing 3 changed files with 842 additions and 0 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -94,6 +94,7 @@
     '**.ipynb_checkpoints',
     # Ignore markdown source for notebooks; myst-nb builds from the ipynb
     'notebooks/deep_learning/*.md',
+    'notebooks/distributed/*.md',
     'notebooks/implicit_diff/*.md']
 
 
@@ -132,6 +133,7 @@
 nb_execution_excludepatterns = [
     # Slow notebook
     'notebooks/deep_learning/*.*',
+    'notebooks/distributed/*.*',
     'notebooks/implicit_diff/dataset_distillation.*',
     'notebooks/implicit_diff/maml.*',
 ]