torch>=2.5
torch_scatter>=2.1
