torch
numpy
tiktoken
tqdm
