matplotlib
numpy
jax
cmastro
optax
equinox
imageio
