torch>=2.4.0

[dev]
pytest

[fa3]
flash_attn_3

[hopper]
torch>=2.8.0
flash_attn_3
