torch>=2.0.0

[all]
transformers>=4.30.0
datasets>=2.0.0
mypy>=1.5
ruff>=0.1

[cuda]
torch>=2.0.0

[dev]
pytest>=7.0
pytest-cov
mypy>=1.5
ruff>=0.1
build
twine

[hf]
transformers>=4.30.0
datasets>=2.0.0
