torch
numpy
transformers
tiktoken
wandb
tqdm
streamlit
subprocess