torch>=2.8
transformers==4.57.6
numpy
safetensors
# eval_decisions.py only (downloads the public eval splits):
datasets>=4.0