torch
numpy
scipy
seaborn
wandb
matplotlib
Pyaml
neuralpredictors==0.1.0
nnfabrik
tqdm