{"library":"numpyro","install":[{"cmd":"pip install numpyro[cuda12_pip] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html","imports":["import numpyro","import numpyro.distributions as dist","from numpyro.infer import MCMC, NUTS","import jax; key = jax.random.PRNGKey(0); key1, key2 = jax.random.split(key)"]},{"cmd":"pip install numpyro[cpu]","imports":[]}]}