torch>=2.0
transformers>=4.40
tensorly>=0.9
tensorly-torch>=0.5
accelerate>=1.0
datasets>=3.0
pyyaml>=6.0
numpy>=1.24
scipy>=1.10
pandas>=2.0
scikit-learn>=1.3
tqdm>=4.60
psutil>=5.9
huggingface-hub>=0.20
click>=8.0
safetensors>=0.4

[cola]
cola-ml

[dev]
pytest>=7.0
pytest-cov>=4.0
ruff>=0.1

[eval]
lm_eval>=0.4
evaluate>=0.4
rouge_score>=0.1
sacrebleu>=2.0
