torch
transformers
torch-scatter