deepspeed
torch
transformers==4.21.2
