torch >= 1.13
transformers >= 4.37, <4.42
datasets
tokenizers >= 0.13.3
accelerate
deepspeed
numpy
scipy
sentencepiece
wandb
tensorboard
optree
matplotlib
tqdm
rich
optree

tenacity
python-dotenv
openai