# Core
jax
jaxlib
equinox
optax

# Extra
matplotlib
jupyter
jupyterlab
tqdm
chainconsumer