requirements.txt

absl-py==2.1.0
arviz==0.18.0
asttokens==2.4.1
chex==0.1.86
comm==0.2.2
contourpy==1.2.1
cycler==0.12.1
debugpy==1.8.1
decorator==5.1.1
dill==0.3.8
dm-tree==0.1.8
etils==1.9.2
executing==2.0.1
flax==0.8.5
fonttools==4.53.0
fsspec==2024.6.1
funsor==0.4.5
h5netcdf==1.3.0
h5py==3.11.0
humanize==4.10.0
importlib_resources==6.4.4
ipykernel==6.29.4
ipython==8.25.0
jax==0.4.31
jax-cuda12-pjrt==0.4.31
jax-cuda12-plugin==0.4.31
jaxlib==0.4.31
jedi==0.19.1
jmp==0.0.4
joblib==1.4.2
jupyter_client==8.6.2
jupyter_core==5.7.2
kiwisolver==1.4.5
makefun==1.15.2
markdown-it-py==3.0.0
matplotlib==3.9.0
matplotlib-inline==0.1.7
mdurl==0.1.2
ml-dtypes==0.4.0
msgpack==1.0.8
multipledispatch==1.0.0
nest-asyncio==1.6.0
numpy==1.26.4
nvidia-cublas-cu12==12.5.2.13
nvidia-cuda-cupti-cu12==12.5.39
nvidia-cuda-nvcc-cu12==12.5.40
nvidia-cuda-nvrtc-cu12==12.4.127
nvidia-cuda-runtime-cu12==12.5.39
nvidia-cudnn-cu12==9.3.0.75
nvidia-cufft-cu12==11.2.3.18
nvidia-cusolver-cu12==11.6.2.40
nvidia-cusparse-cu12==12.4.1.24
nvidia-nccl-cu12==2.21.5
nvidia-nvjitlink-cu12==12.5.40
opt-einsum==3.3.0
optax==0.2.3
orbax-checkpoint==0.6.0
packaging==24.1
pandas==2.2.2
parso==0.8.4
pexpect==4.9.0
pillow==10.3.0
platformdirs==4.2.2
prompt_toolkit==3.0.47
protobuf==5.27.3
psutil==5.9.8
ptyprocess==0.7.0
pure-eval==0.2.2
Pygments==2.18.0
pyparsing==3.1.2
python-dateutil==2.9.0.post0
pytz==2024.1
PyYAML==6.0.2
pyzmq==26.0.3
rich==13.7.1
scikit-learn==1.5.0
scipy==1.13.1
seaborn==0.13.2
setuptools==70.0.0
six==1.16.0
stack-data==0.6.3
tabulate==0.9.0
tensorstore==0.1.64
threadpoolctl==3.5.0
toolz==0.12.1
tornado==6.4.1
tqdm==4.66.4
traitlets==5.14.3
typing_extensions==4.12.2
tzdata==2024.1
wcwidth==0.2.13
xarray==2024.5.0
xarray-einstats==0.7.0
zipp==3.20.0
numpyro @ git+https://github.com/pyro-ppl/numpyro.git@master  #TODO: remove upon release