torch>=2.0.0
triton>=2.0.0

[benchmark]
transformers
rich

[dev]
pytest
rich
