torch>=2.0
numpy
numba
tqdm

[:platform_system == "Linux"]
triton>=3.6
nvidia-cutlass-dsl

[:platform_system == "Windows"]
triton-windows>=3.6

[bench]
scikit-learn<1.8,>=1.5
tqdm
matplotlib

[dev]
pytest>=7
scikit-learn<1.8,>=1.5
tqdm
