arviz==1.0.0
diffrax>=0.7.2
blackjax>=1.3
dm-haiku>=0.0.16
einops>=0.8.2
jrnmm==0.1.1.post2
matplotlib>=3.10.8
optax>=0.2.7
seaborn>=0.13.2
surjectors>=0.3.3
tqdm>=4.64.1
ipython>=9.11.0
jaxlib==0.8.1
jax==0.8.1
grain>=0.2.16
tfp-nightly[jax]>=0.26.0.dev20260318
