jax
jaxlib
numpy
scipy
matplotlib
ipython
jaxopt
tqdm
torch