Skip to content

Commit

Permalink
Link fixing
Browse files Browse the repository at this point in the history
  • Loading branch information
GleasonK committed Apr 19, 2024
1 parent c9b6150 commit 1ee1a24
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 4 deletions.
7 changes: 5 additions & 2 deletions docs/tutorials/jax-export.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -19,13 +19,16 @@
"source": [
"# Tutorial: Exporting StableHLO from JAX\n",
"\n",
"[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/openxla/stablehlo/blob/main/docs/tutorials/jax-export.ipynb) [![Open in Kaggle](https://kaggle.com/static/images/open-in-kaggle.svg)](https://kaggle.com/kernels/welcome?src=https://github.com/openxla/stablehlo/blob/main/docs/tutorials/jax-export.ipynb)\n",
"[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)][jax-tutorial-colab] [![Open in Kaggle](https://kaggle.com/static/images/open-in-kaggle.svg)][jax-tutorial-kaggle]\n",
"\n",
"## Tutorial Setup\n",
"\n",
"### Install required dependencies\n",
"\n",
"We'll be using `jax` and `jaxlib` (JAX's XLA package), along with `flax` and `transformers` for some models to export."
"We'll be using `jax` and `jaxlib` (JAX's XLA package), along with `flax` and `transformers` for some models to export.",
"\n",
"[jax-tutorial-colab]: https://colab.research.google.com/github/openxla/stablehlo/blob/main/docs/tutorials/jax-export.ipynb\n",
"[jax-tutorial-kaggle]: https://kaggle.com/kernels/welcome?src=https://github.com/openxla/stablehlo/blob/main/docs/tutorials/jax-export.ipynb"
],
"metadata": {
"id": "_TuAgGNKt5HO"
Expand Down
8 changes: 6 additions & 2 deletions docs/tutorials/pytorch-export.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -19,15 +19,19 @@
"source": [
"# Tutorial: Exporting StableHLO from PyTorch\n",
"\n",
"[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/openxla/stablehlo/blob/main/docs/tutorials/pytorch-export.ipynb) [![Open in Kaggle](https://kaggle.com/static/images/open-in-kaggle.svg)](https://kaggle.com/kernels/welcome?src=https://github.com/openxla/stablehlo/blob/main/docs/tutorials/pytorch-export.ipynb)\n",
"[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)][pytorch-tutorial-colab] [![Open in Kaggle](https://kaggle.com/static/images/open-in-kaggle.svg)][pytorch-tutorial-kaggle]\n",
"[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)]() ",
"\n",
"_Intro to the [`torch_xla.stablehlo`](https://github.com/pytorch/xla/blob/main/docs/stablehlo.md) module._\n",
"\n",
"## Tutorial Setup\n",
"\n",
"### Install required dependencies\n",
"\n",
"We'll be using `torch` and `torchvision` to get a `resnet18` model, and `torch_xla` to export to StableHLO."
"We'll be using `torch` and `torchvision` to get a `resnet18` model, and `torch_xla` to export to StableHLO.",
"\n",
"[pytorch-tutorial-colab]: https://colab.research.google.com/github/openxla/stablehlo/blob/main/docs/tutorials/pytorch-export.ipynb\n",
"[pytorch-tutorial-kaggle]: https://kaggle.com/kernels/welcome?src=https://github.com/openxla/stablehlo/blob/main/docs/tutorials/pytorch-export.ipynb"
],
"metadata": {
"id": "42_6VTj7l_xS"
Expand Down

0 comments on commit 1ee1a24

Please sign in to comment.