Simplify GPU admonition and add to all JAX lectures

- Simplify _admonition/gpu.md text (more concise messaging)
- Add GPU admonition include to 17 JAX lectures that were missing it

JAX lectures now with GPU admonition (21 total):
- aiyagari.md, ak_aiyagari.md, ar1_bayes.md, back_prop.md
- bayes_nonconj.md, endogenous_lake.md, ifp_advanced.md
- ifp_discrete.md, ifp_egm.md, ifp_egm_transient_shocks.md
- ifp_opi.md, lake_model.md, mccall_fitted_vfi.md
- mccall_model.md, mccall_model_with_sep_markov.md
- mccall_model_with_separation.md, mccall_persist_trans.md
- mix_model.md, mle.md, newton_method.md, os_egm_jax.md

Author: mmcky

lectures/_admonition/gpu.md

@@ -1,9 +1,7 @@
 ```{admonition} GPU
 :class: warning
 
-This lecture was built using a machine with the latest CUDA and CUDANN frameworks installed with access to a GPU.
+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.
 
-To run this lecture on [Google Colab](https://colab.research.google.com/), click on the "play" icon top right, select Colab, and set the runtime environment to include a GPU.
-
-To run this lecture on your own machine, you need to install the software listed following this notice.
+For local GPU setup, see the [JAX installation guide](https://github.com/google/jax).
 ```

lectures/aiyagari.md

@@ -22,6 +22,9 @@ kernelspec:
 
 # The Aiyagari Model
 
+```{include} _admonition/gpu.md
+```
+
 ```{contents} Contents
 :depth: 2
 ```

lectures/ak_aiyagari.md

@@ -14,6 +14,9 @@ kernelspec:
 
 # A Long-Lived, Heterogeneous Agent, Overlapping Generations Model
 
+```{include} _admonition/gpu.md
+```
+
 In addition to what's in Anaconda, this lecture will need the following library
 
 ```{code-cell} ipython3

lectures/endogenous_lake.md

@@ -22,6 +22,9 @@ kernelspec:
 
 # Lake Model with an Endogenous Job Finding Rate
 
+```{include} _admonition/gpu.md
+```
+
 ```{index} single: Lake Model, Endogenous
 ```
 

lectures/ifp_advanced.md

@@ -19,6 +19,9 @@ kernelspec:
 
 # {index}`The Income Fluctuation Problem V: Stochastic Returns on Assets <single: The Income Fluctuation Problem V: Stochastic Returns on Assets>`
 
+```{include} _admonition/gpu.md
+```
+
 ```{contents} Contents
 :depth: 2
 ```

lectures/ifp_discrete.md

@@ -13,6 +13,8 @@ kernelspec:
 
 # The Income Fluctuation Problem I: Discretization and VFI
 
+```{include} _admonition/gpu.md
+```
 
 ## Overview
 

lectures/ifp_egm.md

@@ -21,6 +21,9 @@ kernelspec:
 
 # {index}`The Income Fluctuation Problem III: The Endogenous Grid Method <single: The Income Fluctuation Problem III: The Endogenous Grid Method>`
 
+```{include} _admonition/gpu.md
+```
+
 ```{contents} Contents
 :depth: 2
 ```

lectures/ifp_egm_transient_shocks.md

@@ -21,6 +21,9 @@ kernelspec:
 
 # {index}`The Income Fluctuation Problem IV: Transient Income Shocks <single: The Income Fluctuation Problem IV: Transient Income Shocks>`
 
+```{include} _admonition/gpu.md
+```
+
 ```{contents} Contents
 :depth: 2
 ```

lectures/ifp_opi.md

@@ -13,6 +13,8 @@ kernelspec:
 
 # The Income Fluctuation Problem II: Optimistic Policy Iteration
 
+```{include} _admonition/gpu.md
+```
 
 ## Overview
 

lectures/lake_model.md

@@ -22,6 +22,9 @@ kernelspec:
 
 # A Lake Model of Employment and Unemployment
 
+```{include} _admonition/gpu.md
+```
+
 ```{index} single: Lake Model
 ```