torch>=1.12
numpy
pandas
scikit-learn
matplotlib
PyWavelets
pytorch_wavelets
wandb
