torch
transformers
scipy