torch==2.1.1
torchvision==0.16.1
torchaudio==2.1.1
lightning==2.2.4
numpy==1.26.4
tqdm==4.66.4
pyvista==0.44.0
wandb==0.19.8
scikit-image==0.23.2
scikit-learn==1.4.2
scipy==1.13.0
timm==1.0.7
einops==0.8.0
h5py==3.11.0
matplotlib==3.9.0
networkx==3.2.1
torchdata==0.7.1
pandas
dgl