fire
numpy
pandas
tiktoken
torch
tqdm
wandb
matplotlib
scikit-learn
regex
datasets