torch
numpy
transformers
datasets
tiktoken
wandb
tqdm