torch>=2.0.0
numpy>=1.20.0

[all]
triton>=2.0.0
flash-attn>=2.0.0
pytest>=7.0.0
pytest-cov>=4.0.0

[gpu]
triton>=2.0.0
flash-attn>=2.0.0

[tests]
pytest>=7.0.0
pytest-cov>=4.0.0
