diff --git a/.github/workflows/cache.yml b/.github/workflows/cache.yml index 2baed89a..981325bf 100644 --- a/.github/workflows/cache.yml +++ b/.github/workflows/cache.yml @@ -21,7 +21,13 @@ jobs: - name: Install JAX and Numpyro shell: bash -l {0} run: | - pip install -U "jax[cuda13]" + # Pinned: jax 0.11.1 (2026-08-17) hangs numpy_vs_numba_vs_jax's + # lax.fori_loop cells -- they jit with device=cpu, so the CPU-path + # regression bites even on these GPU runners. Context is recorded in + # QuantEcon/workspace-lectures#49. With jax already present, the + # lecture's unpinned `!pip install quantecon jax` cell is satisfied + # and does not upgrade. + pip install "jax[cuda13]==0.11.0" pip install numpyro python scripts/test-jax-install.py - name: Check nvidia drivers diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index b36ab89d..e5ed1e84 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -26,7 +26,13 @@ jobs: - name: Install JAX and Numpyro shell: bash -l {0} run: | - pip install -U "jax[cuda13]" + # Pinned: jax 0.11.1 (2026-08-17) hangs numpy_vs_numba_vs_jax's + # lax.fori_loop cells -- they jit with device=cpu, so the CPU-path + # regression bites even on these GPU runners. Context is recorded in + # QuantEcon/workspace-lectures#49. With jax already present, the + # lecture's unpinned `!pip install quantecon jax` cell is satisfied + # and does not upgrade. + pip install "jax[cuda13]==0.11.0" pip install numpyro python scripts/test-jax-install.py - name: Install latex dependencies diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml index e61eb6d6..f832593d 100644 --- a/.github/workflows/publish.yml +++ b/.github/workflows/publish.yml @@ -38,7 +38,13 @@ jobs: - name: Install JAX and Numpyro shell: bash -l {0} run: | - pip install -U "jax[cuda13]" + # Pinned: jax 0.11.1 (2026-08-17) hangs numpy_vs_numba_vs_jax's + # lax.fori_loop cells -- they jit with device=cpu, so the CPU-path + # regression bites even on these GPU runners. Context is recorded in + # QuantEcon/workspace-lectures#49. With jax already present, the + # lecture's unpinned `!pip install quantecon jax` cell is satisfied + # and does not upgrade. + pip install "jax[cuda13]==0.11.0" pip install numpyro python scripts/test-jax-install.py - name: Check nvidia drivers