numpy==2.2.4
torch==2.6.0
transformers>=4.51.0
datasets==3.4.1
lm-eval==0.3.0
accelerate>=1.6.0
tensorboardX
bitsandbytes
sentencepiece
tqdm
pytorch-optimizer>=3.5.0
GPUtil>=1.4.0
mpi4py
