torch
numpy
transformers
datasets
matplotlib
tqdm
scikit-learn
openai