torch
timm
numpy
torch_scatter
torch_geometric
