torch
transformers==4.31.0
tensorly
tokenizers
datasets
peft==0.10.0
wandb
loguru
nvitop
lion-pytorch
matplotlib
bitsandbytes
scipy
scikit-learn
evaluate
importlib_metadata
lm_eval
