jax>=0.3.17
jaxlib>=0.3.14
tensorflow
imageio
jupyter
matplotlib
numpy<2
pandas
igraph
scikit-learn
tqdm
dm-haiku>=0.0.8
psutil
optax
pyarrow==7.0.0
tensorflow-datasets~=4.3.0
deepdiff
huggingface-hub
