Remove PyTorch and Pyro-PPL from workflows

.github/workflows/cache.yml

@@ -20,12 +20,11 @@ jobs:
           python-version: "3.13"
           environment-file: environment.yml
           activate-environment: quantecon
-      - name: Install JAX, Numpyro, PyTorch
+      - name: Install JAX and Numpyro
         shell: bash -l {0}
         run: |
-          pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
-          pip install --upgrade "jax[cuda12-local]==0.6.2"
-          pip install numpyro  
+          pip install -U "jax[cuda13]"
+          pip install numpyro
           python scripts/test-jax-install.py
       - name: Check nvidia drivers
         shell: bash -l {0}

.github/workflows/ci.yml

@@ -28,17 +28,15 @@ jobs:
           python-version: "3.13"
           environment-file: environment.yml
           activate-environment: quantecon
-      - name: Install JAX, Numpyro, PyTorch
-        shell: bash -l {0}
-        run: |
-          # pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
-          # pip install pyro-ppl
-          pip install "jax[cuda12-local]==0.6.2"
-          pip install numpyro pyro-ppl
-          python scripts/test-jax-install.py
       - name: Check nvidia Drivers
         shell: bash -l {0}
         run: nvidia-smi
+      - name: Install JAX and Numpyro
+        shell: bash -l {0}
+        run: |
+          pip install -U "jax[cuda13]"
+          pip install numpyro
+          python scripts/test-jax-install.py
       - name: Display Conda Environment Versions
         shell: bash -l {0}
         run: conda list

.github/workflows/publish.yml

@@ -19,12 +19,11 @@ jobs:
           python-version: "3.13"
           environment-file: environment.yml
           activate-environment: quantecon
-      - name: Install JAX, Numpyro, PyTorch
+      - name: Install JAX and Numpyro
         shell: bash -l {0}
         run: |
-          pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
-          pip install --upgrade "jax[cuda12-local]==0.6.2"
-          pip install numpyro  
+          pip install -U "jax[cuda13]"
+          pip install numpyro
           python scripts/test-jax-install.py
       - name: Check nvidia drivers
         shell: bash -l {0}