torch
tiktoken
numpy
psutil
tqdm
