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

remove cpu platform setting in quickstart

Browse Source 
fixes #9244
This commit is contained in:
Matthew Johnson 2022-01-19 10:24:09 -08:00
parent 98b5b3a2a9
commit 726f60f6bc
2 changed files with 0 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
docs/notebooks/quickstart.ipynb
View File

@ -43,20 +43,6 @@
"from jax import random"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"tags": [
"remove-cell"
]
},
"outputs": [],
"source": [
"# Prevent GPU/TPU warning.\n",
"import jax; jax.config.update('jax_platform_name', 'cpu')"
]
},
{
"cell_type": "markdown",
"metadata": {

7
docs/notebooks/quickstart.md
View File

@ -45,13 +45,6 @@ from jax import grad, jit, vmap
from jax import random
```
```{code-cell} ipython3
:tags: [remove-cell]
# Prevent GPU/TPU warning.
import jax; jax.config.update('jax_platform_name', 'cpu')
```
+++ {"id": "FQ89jHCYfhpg"}
## Multiplying Matrices