numba
numpy
torch
tqdm
more-itertools
tiktoken==0.3.3