pip jax for testing

Author: aloctavodia

.azure-pipelines/azure-pipelines-external.yml

@@ -45,7 +45,7 @@ jobs:
       fi
 
       python -m pip --no-cache-dir install emcee>=3 h5py
-      python -m pip --no-cache-dir install pystan pyro-ppl>=1 numpyro jax
+      python -m pip --no-cache-dir install pystan pyro-ppl>=1 numpyro jax<0.10.0
 
     displayName: 'Install external dependencies'