jax
equinox
numpy
matplotlib
tqdm
poppy
