torch
scipy
numpy
transformers