transformers
datasets
torch
torchdata==0.9.0
deepspeed
pandas
swanlab