torch>=2.7.0
transformers>=4.20.0
numpy>=1.21.0
pandas>=1.3.0
tqdm>=4.62.0
matplotlib>=3.4.0
seaborn>=0.11.0
datasets>=2.0.0
accelerate>=0.12.0
sacrebleu>=2.0.0
sqlitedict>=2.0.0
lm-eval>=0.4.0

[cuda]
torch>=2.7.0
torchvision>=0.19.0
torchaudio>=2.4.0

[cuda128]

[cuda128:platform_system != "Darwin"]
torch>=2.7.0
torchvision>=0.19.0
torchaudio>=2.4.0

[dev]
pytest>=6.0
pytest-cov>=2.0
black>=21.0
flake8>=3.8
isort>=5.0
