numpy
torch
pytorch-lightning
pandas
tqdm
PyYAML
matplotlib
drms
astropy<8,>=7.2
sunpy<8,>=6
h5netcdf
h5py
itipy
aiapy<0.11,>=0.6.4

[evaluation]
scikit-learn
seaborn
plotly

[training]
wandb
torchvision
