torch>=2.1.2
triton>=2.3.0
transformers>=4.42.0

[dev]
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
