numpy
scipy
statsmodels
tqdm

[cpu]
torch>=2.0

[cuda]
torch>=2.0
triton>=2.0

[cupy]
cupy-cuda13x>=13.0

[dev]
pytest
torch>=2.0
triton>=2.0
scikit-learn

[kmeans]
scikit-learn
