torch>=2.0.0
transformers>=4.38.0
numpy>=1.24.0
datasets>=2.14.0
tqdm>=4.66.0
psutil>=5.9.0
safetensors>=0.4.0
accelerate>=0.26.0

[dev]
pytest>=7.4
pytest-cov>=4.1
black>=23.0
isort>=5.12
mypy>=1.5
ruff>=0.1.0
pre-commit>=3.4

[eval]
lm-eval>=0.4.0

[triton]
triton>=2.1.0
