Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions lectures/_admonition/gpu.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
```{admonition} GPU
:class: warning

This lecture is designed to run on a GPU. To use Google Colab's free GPUs, click the play icon top right, select Colab, and set the runtime to include a GPU. For local GPU setup, see the [JAX installation guide](https://github.com/google/jax).
```
11 changes: 1 addition & 10 deletions lectures/jax_intro.md
Original file line number Diff line number Diff line change
Expand Up @@ -33,16 +33,7 @@ In addition to what's in Anaconda, this lecture will need the following librarie
!pip install jax quantecon
```

```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and targets JAX for GPU programming.

Free GPUs are available on Google Colab.
To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support.
If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```{include} _admonition/gpu.md
```

## JAX as a NumPy Replacement
Expand Down
11 changes: 1 addition & 10 deletions lectures/numpy_vs_numba_vs_jax.md
Original file line number Diff line number Diff line change
Expand Up @@ -48,16 +48,7 @@ tags: [hide-output]
!pip install quantecon jax
```

```{admonition} GPU
:class: warning

This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and target JAX for GPU programming.

Free GPUs are available on Google Colab.
To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.

Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support.
If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
```{include} _admonition/gpu.md
```

We will use the following imports.
Expand Down
Loading