numpy
torch
transformers
scikit-learn
pandas
thop
pyJoules
