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

Merge pull request #19437 from 8bitmp3:shard_map_title

Browse Source 
PiperOrigin-RevId: 600326508
This commit is contained in:
jax authors 2024-01-21 20:00:56 -08:00
commit a1213c7dd9
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/notebooks/shard_map.ipynb
View File

@ -5,7 +5,7 @@
"id": "97c57a94",
"metadata": {},
"source": [
"# Intro\n",
"# SPMD multi-device parallelism with `shard_map`\n",
"\n",
"`shard_map` is a single-program multiple-data (SPMD) multi-device parallelism API to map a function over shards of data. Mapped function applications, or _instances_, communicate with each other via explicit collective communication operations.\n",
"\n",

2
docs/notebooks/shard_map.md
View File

@ -14,7 +14,7 @@ kernelspec:
name: python3
---
# Intro
# SPMD multi-device parallelism with `shard_map`
`shard_map` is a single-program multiple-data (SPMD) multi-device parallelism API to map a function over shards of data. Mapped function applications, or _instances_, communicate with each other via explicit collective communication operations.