torch
numpy
transformers
wandb
tqdm
streamlit