numpy
torch>=1.0
trl
scikit-learn
imbalanced-learn
matplotlib
seaborn
scipy
transformers
torchvision
tqdm
