Skip to content

Commit e8f43d1

Browse files
committed
Explicit sharding docs
1 parent 8fbe3b1 commit e8f43d1

File tree

4 files changed

+1132
-0
lines changed

4 files changed

+1132
-0
lines changed

docs/advanced_guide.rst

+1
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ operations.
1212
:maxdepth: 1
1313

1414
notebooks/Distributed_arrays_and_automatic_parallelization
15+
notebooks/explicit-sharding
1516
notebooks/shard_map
1617
multi_process
1718
distributed_data_loading

docs/conf.py

+1
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,7 @@ def _do_not_evaluate_in_jax(
222222
'jep/9407-type-promotion.*',
223223
# TODO(jakevdp): enable execution on the following if possible:
224224
'notebooks/Distributed_arrays_and_automatic_parallelization.*',
225+
'notebooks/explicit-sharding.*',
225226
'notebooks/autodiff_remat.*',
226227
# Fails on readthedocs with Kernel Died
227228
'notebooks/convolutions.ipynb',

0 commit comments

Comments
 (0)