tensorflow
jax
jaxlib
flax
numpy
networkx
tqdm
scipy
