torch>=2.1.2
triton>=2.3.1

[amd]
torch>=2.6.0.dev
setuptools-scm>=8
torchvision>=0.20.0.dev
triton>=3.0.0

[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
torchvision>=0.16.2
seaborn

[transformers]
transformers~=4.0

[trl]
trl>=0.11.0
