numpy
scipy
jupyter
ipykernel
matplotlib
h5py
nibabel
tqdm
pandas
pycocotools
requests
torch==2.4.0
diffusers==0.30.2
transformers==4.44.2
einops
scikit-learn
scikit-image
visualpriors
pytorch-lightning
wandb
timm==0.6.7