diffrax>=0.7.2
distrax>=0.1.9
blackjax>=1.3
dm-haiku>=0.0.16
einops>=0.8.2
optax>=0.2.8
surjectors>=0.3.3
tqdm>=4.64.1
jaxlib>=0.10.1
jax>=0.10.1
grain>=0.2.16
tfp-nightly[jax]>=0.26.0.dev20260318

[all]
jrnmm==0.1.1.post2
