torch>=2.0

[accelerate]
accelerate>=0.28

[all]
transformers>=4.40
accelerate>=0.28
lightning>=2.1

[dev]
pytest>=7.0
pytest-timeout>=2.0
ruff>=0.4

[hf]
transformers>=4.40

[lightning]
lightning>=2.1

[release]
build>=1.2.0
twine>=5.0.0
