torch>=2.0.0
numpy
tiktoken
wandb
datasets
tqdm