numpy
flax
jax
numpy
numpyro
optax
matplotlib
pyswarms
scipy
tqdm
