ipdb
matplotlib
seaborn
pandas
tqdm
jupyter
equinox
optax
einops
scikit-learn
tensorflow
tensorflow-datasets
hydra-core
hydra-submitit-launcher
wandb
