rapidsai/cuml

cuML is a GPU-accelerated machine‑learning library that forms part of the RAPIDS suite, offering fast, scalable implementations of common algorithms such as lin