colossalai
evaluate
datasets
torch
tqdm
transformers
scipy
scikit-learn
ptflops