diff --git a/lectures/_admonitions/gpu.md b/lectures/_admonitions/gpu.md new file mode 100644 index 000000000..a955930a1 --- /dev/null +++ b/lectures/_admonitions/gpu.md @@ -0,0 +1,9 @@ +```{admonition} GPU +:class: warning + +This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and 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]` +``` \ No newline at end of file diff --git a/lectures/kesten_processes.md b/lectures/kesten_processes.md index d8e2d7101..5b09cffc1 100644 --- a/lectures/kesten_processes.md +++ b/lectures/kesten_processes.md @@ -19,14 +19,7 @@ kernelspec: # Kesten Processes and Firm Dynamics -```{admonition} GPU -:class: warning - -This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and 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} _admonitions/gpu.md ``` ```{index} single: Linear State Space Models diff --git a/lectures/wealth_dynamics.md b/lectures/wealth_dynamics.md index cc65ea062..59ae4bd9d 100644 --- a/lectures/wealth_dynamics.md +++ b/lectures/wealth_dynamics.md @@ -19,14 +19,7 @@ kernelspec: # Wealth Distribution Dynamics -```{admonition} GPU -:class: warning - -This lecture is accelerated via [hardware](status:machine-details) that has access to a GPU and 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} _admonitions/gpu.md ``` ```{contents} Contents