numpy
scipy
jax
jaxopt
flax
tqdm
matplotlib
pmlb
