torch 
transformers 
wandb
datasets
accelerate>=0.26.0
deepspeed