requirements.txt

aenum
argparse
#ast
#collections
#ConfigSpace  # for hpo, install issues...
#importlib
#inspect
#math
matplotlib
more_itertools  # for hpo
networkx
numba
numpy
#os
optuna # for hpo
pandas
protobuf
psutil
pytorch-lightning
pyyaml
scikit-learn
scipy
torch
#torch-scatter  # has to be installed separately via: ...> pip install torch-scatter -f https://data.pyg.org/whl/torch-2.3.0+${CUDA}.html with '${CUDA}' = 'cpu', 'cu118' or 'cu121' depending on PyTorch installation (usually 'cpu')
torchvision
tqdm
#typing  # optional?
wandb  # for hpo