Skip to content

Conversation

@mmcky
Copy link
Contributor

@mmcky mmcky commented Nov 30, 2025

Summary

This PR simplifies the GPU admonition text and ensures all JAX lectures include the GPU notice.

Changes

1. Simplified GPU Admonition

Updated lectures/_admonition/gpu.md from:

This lecture was built using a machine with the latest CUDA and CUDANN frameworks 
installed with access to a GPU.

To run this lecture on Google Colab, 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.

To:

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.

2. Added GPU Admonition to 17 JAX Lectures

The following lectures were missing the GPU admonition and now include it:

  • aiyagari.md
  • ak_aiyagari.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
  • mle.md
  • newton_method.md
  • os_egm_jax.md

Benefits

- 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
@mmcky mmcky merged commit 0bfcac8 into main Nov 30, 2025
1 check passed
@mmcky mmcky deleted the simplify-gpu-admonition branch November 30, 2025 23:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants