mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

[shard-map] add user tutorial

Browse Source
This commit is contained in:
Matthew Johnson 2024-01-08 13:12:04 -08:00
parent ba10775eda
commit 8b219d5f7b
4 changed files with 3339 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/advanced_guide.rst
View File

@ -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::

1
docs/conf.py
View File

@ -209,6 +209,7 @@ nb_execution_excludepatterns = [
'notebooks/xmap_tutorial.*',
'notebooks/Distributed_arrays_and_automatic_parallelization.*',
'notebooks/autodiff_remat.*',
'notebooks/shard_map.*',
# Requires accelerators
'pallas/quickstart.*',
]

1964
docs/notebooks/shard_map.ipynb Normal file
View File

File diff suppressed because one or more lines are too long

1373
docs/notebooks/shard_map.md Normal file
View File

File diff suppressed because one or more lines are too long