torch>=2.1
tqdm>=4

[attribution]
dattri>=0.3.0

[dev]
ruff==0.15.20
pre-commit>=3
pytest>=8
dattri>=0.3.0
transformers>=4.40
accelerate>=0.28
trl>=0.19
ai2-olmo

[olmo]
ai2-olmo

[olmo-core]
ai2-olmo-core==2.6.0

[test]
pytest>=8

[transformers]
transformers>=4.40
accelerate>=0.28

[trl]
trl>=0.19
