
[bench]
flash-attn>=2.6
matplotlib>=3.7

[runtime]
torch>=2.4
triton>=3.0

[test]
pytest>=7
