torch
transformers>=4.24.0
accelerate>=0.14.0
evaluate
datasets>=2.6.1
einops==0.6.0
numpy
tqdm
setuptools
click
wandb
seaborn
matplotlib
plotly
zstandard
pyrallis
scipy
scikit-learn
gdown
fast-bleu
mauve-text
lion-pytorch
lovely-tensors