jax
jaxlib
zodiax==0.3.3
optax
matplotlib
jupyter
jupyterlab
tqdm
chainconsumer
