numpy>=1.13.3
torch>=0.4.0
transformers>=3.5.1,<4.0.0
sklearn
