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

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

View File

@ -876,6 +876,16 @@
"execution_count": 0,
"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",
"metadata": {
@ -1001,4 +1011,4 @@
"outputs": []
}
]
}
}

Binary file not shown.

After

Width:  |  Height:  |  Size: 23 KiB