numpy
torch==1.13.1
torchvision
matplotlib
scipy
h5py
pandas
kornia
wandb
pytorch-lightning==1.5.10
seaborn
scikit-learn
imageio
imageio-ffmpeg
gradio==3.40.0