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
pytest-asyncio
pytest-rerunfailures
datasets>=2.19.2
seaborn
mkdocs-material
torchvision>=0.20
prek>=0.2.28
