torch>=2.0.0
wandb
datasets
numpy
tqdm
matplotlib
safetensors