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

Merge pull request #9245 from mattjj:issue9244

Browse Source 
PiperOrigin-RevId: 422855209
This commit is contained in:
jax authors 2022-01-19 11:00:00 -08:00
commit e152000c03
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" "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", "cell_type": "markdown",
"metadata": { "metadata": {

7
docs/notebooks/quickstart.md
View File

@ -45,13 +45,6 @@ from jax import grad, jit, vmap
from jax import random 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"} +++ {"id": "FQ89jHCYfhpg"}
## Multiplying Matrices ## Multiplying Matrices