torch==2.1.2
pytorch_lightning==1.9.5
deepspeed==0.12.6