numpy
jax
jaxopt
optax
scipy
tqdm
