numpy
torch>=1.10.0
torchdrug
torch-scatter>=2.0.8
pandas<=1.3.5
scipy
tqdm
six
