torch>=2.1.0
transformers>=4.40.0
numpy>=1.24.0
matplotlib>=3.7.0

[cuda]

[cuda:sys_platform != "win32"]
triton>=2.1.0

[cuda:sys_platform == "win32"]
triton-windows>=3.1.0

[dev]
pytest>=7.4.0
