torch>=2.1.2
triton>=2.3.1

[dev]
transformers>=4.44.2
matplotlib>=3.7.2
flake8>=4.0.1.1
black>=24.4.2
isort>=5.13.2
pytest>=7.1.2
pytest-xdist
pytest-rerunfailures
datasets>=2.19.2
seaborn
mkdocs
mkdocs-material
torchvision>=0.20
