numpy==1.26.4
torch
transformers
deepspeed
numerize
accelerate
datasets
wandb
matplotlib
sentencepiece
peft
