AI Potluck
Back to Gap Map Infrastructure / Classic ML & computer vision

cuML

NVIDIA
open source / Overall score: 3.0

NVIDIA's GPU-accelerated machine-learning library in the RAPIDS suite, with scikit-learn-style estimators for clustering, regression, classification, dimensionality reduction, nearest neighbors and time series - more than fifty algorithms, including GPU versions of UMAP and HDBSCAN. Its cuml.accel mode runs existing scikit-learn, UMAP and HDBSCAN code on the GPU with no code changes, and it scales across GPUs and nodes with Dask.

Openness

5 medium confidence
5.0
license
Apache-2.0(OSI)
source
public(github.com/NVIDIA/cuml)
core features withheld
no — no paid edition of cuML found

Apache-2.0 and built from the public repository, which moved from the rapidsai account to NVIDIA's. NVIDIA describes it as an open-source library, and its enterprise software page does not list cuML among paid components. It needs an NVIDIA GPU to run.

Adoption

3 medium confidence
3.0

Monthly PyPI downloads of the cuml-cu12 and cuml-cu13 wheels, summed. RAPIDS is also installed through conda and NVIDIA's own package index, which are not counted, so this is a floor.

Capability

3 high confidence
3.0

cuML covers much of scikit-learn's catalog and runs it on GPUs, often many times faster, which is a difference of speed and scale rather than of how much of the task it does. Choosing and tuning the model stays with the user.

  • https://docs.rapids.ai/api/cuml/stable/ recorded 2026-09-27

    Docs landing: cuML "supports 50+ algorithms across all major machine learning categories, including clustering, regression, classification, dimensionality reduction, and time series analysis", with multi-GPU and multi-node support and cuml.accel for running scikit-learn code on GPUs.

Verified 2026-09-27