Update CI workflow to install only PyTorch packages

Removed 'torchaudio' from the installation command for PyTorch.

Author: jstac

.github/workflows/ci.yml

@@ -31,7 +31,7 @@ jobs:
       - name: Install JAX, Numpyro, PyTorch
         shell: bash -l {0}
         run: |
-          pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
+          pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
           pip install pyro-ppl
           pip install --upgrade "jax[cuda12-local]==0.6.2"
           pip install numpyro pyro-ppl