torch>=2.1.2
triton>=2.3.1

[dev]
transformers>=4.52.0
matplotlib>=3.7.2
ruff>=0.12.0
pytest>=7.1.2
pytest-xdist
pytest-cov
datasets>=2.19.2
