torch>=2.2.0
numpy>=1.26.0

[dev]
pytest>=8.0.0
matplotlib>=3.8.0
datasets>=2.18.0
transformers>=4.40.0

[triton]
triton>=2.2.0
