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

[transformers]
transformers~=4.0
