We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5641c5f commit 1f84a06Copy full SHA for 1f84a06
.github/workflows/ci.yml
@@ -22,6 +22,7 @@ jobs:
22
shell: bash -l {0}
23
run: |
24
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
25
+ pip install pyro-ppl
26
pip install --upgrade "jax[cuda12-local]"
27
pip install numpyro
28
python scripts/test-jax-install.py
0 commit comments