lightning
deepspeed
wandb
'accelerate>=0.26.0'
peft
transformers==4.51.3
bitsandbytes
attrdict
einops
timm
mosaicml-streaming==0.7.2
sentencepiece==0.2.0
scikit-learn
datasets
rouge-score
sacrebleu
unbabel-comet