torch
transformers
scipy
