torch
transformers
scipy
numpy