torch
datasets
matplotlib
tqdm
nltk
safetensors