torch>=2.0
transformers>=4.40,<5.0
accelerate>=0.26
datasets>=2.19
scikit-learn>=1.3
numpy>=1.24
safetensors>=0.4

# optional: only needed with --wandb_project
# wandb