numpy
scipy
matplotlib
tqdm
jupyterlab
ipympl
jax>=0.3.4
jaxlib
dm-haiku
optax
tikzplotlib
