matplotlib
numpy
pandas
python-dotenv
scikit-learn
tokenizers
torch
tqdm
transformers
wandb
