torch>=2.1.0
transformers>=4.36.0
numpy>=1.24.0

[bench]
datasets>=2.15.0
accelerate>=0.25.0
psutil>=5.9.0

[dev]
pytest>=7.4.0
pytest-cov>=4.1.0
pytest-benchmark>=4.0.0
ruff>=0.1.0
mypy>=1.7.0
pre-commit>=3.5.0
build>=1.0.0
twine>=4.0.0
