torch
pytorch-lightning==1.9.5
deepspeed
