numpy
scipy<=1.12.0
jax
flowjax
tqdm
equinox
