numpy
transformers
torch
joblib