jax>=0.3.17
jaxlib>=0.3.14
imageio
jupyter
matplotlib
numpy
scipy
pandas
igraph
scikit-learn
sklearn
tqdm>=4.64.1
dm-haiku>=0.0.8
psutil
optax>=0.1.3
pyarrow==7.0.0
tensorflow-datasets~=4.3.0
deepdiff
seaborn
torch
huggingface-hub
