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

Add nested pmap image to demo notebook. (#2292)

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2020-02-22 23:04:34 -08:00 committed by GitHub
parent 80abdf0c53
commit 7885b1d034
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
cloud_tpu_colabs/NeurIPS_2019_JAX_demo.ipynb
View File

@ -876,6 +876,16 @@
"execution_count": 0, "execution_count": 0,
"outputs": [] "outputs": []
}, },
{
"cell_type": "markdown",
"metadata": {
"id": "f-FBsWeo1AXE",
"colab_type": "text"
},
"source": [
"![nested pmap example](https://raw.githubusercontent.com/google/jax/master/cloud_tpu_colabs/images/nested_pmap.png)"
]
},
{ {
"cell_type": "markdown", "cell_type": "markdown",
"metadata": { "metadata": {
@ -1001,4 +1011,4 @@
"outputs": [] "outputs": []
} }
] ]
} }

BIN
cloud_tpu_colabs/images/nested_pmap.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 23 KiB