torch>=2.1.0
triton>=3.0.0

[dev]
pytest>=7.0
