torch>=2.2.0

[dev]
pytest>=8.0.0
pytest-cov>=5.0.0
ruff>=0.5.0
matplotlib>=3.8.0
datasets>=2.18.0
transformers>=4.40.0

[triton]

[triton:sys_platform == "linux"]
triton>=2.2.0

[triton:sys_platform == "win32"]
triton-windows>=3.6.0.post26
