You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and targets JAX for GPU programming.
5
-
6
-
Free GPUs are available on Google Colab.
7
-
To use this option, please click on the play icon top right, select Colab, and set the runtime environment to include a GPU.
8
-
9
-
Alternatively, if you have your own GPU, you can follow the [instructions](https://github.com/google/jax) for installing JAX with GPU support.
10
-
If you would like to install JAX running on the `cpu` only you can use `pip install jax[cpu]`
4
+
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).
0 commit comments