Registry / ai-ml / raft-dask-cu12

raft-dask-cu12

JSON →
library26.6.0pypypiunverified

RAFT (Reusable Accelerated Functions & Tools) provides GPU-accelerated primitives for machine learning, linear algebra, and spatial computations. The Dask integration enables distributed GPU workflows via Dask. Current version: 26.4.0 (stable). RAPIDS releases every 3 months (quarterly). CUDA 12.x only.

ai-mldatadatabase
pip install raft-dask-cu12
github ↗package ↗homepage ↗docs ↗
INSTALL
IMPORT
SIG · RAFT-DASK-CU12
R
raft-dask-cu12
ai-mlpythonv26.6.0
Install
55.9s avg
Import
—
Disk
2949MB
Pass rate
5/ 10
Env Coverage5 / 10
glibc
3.9–3.13
musl
3.9–3.13
Install & Compatibility
Where this runs
tested against v26.2.0 · pip install
no network on importno background threads
Install × environment matrix
Each cell = how many times install + import succeeded across repeated harness runs. Partial = flaky.
glibc = Debian/Ubuntu slim · musl = Alpine Linux
musl
py 3.10–3.95 runs
build_error
glibc
py 3.10–3.95 runs
installs and imports cleanly · install 55.9s · import 0.000s · 3379.2MB
2949MB installed
● package 2949MB
Code
Verified usage

Verified import paths — ran on the pinned version, not inferred.

DaskClusterProxy
✓ from raft_dask import DaskClusterProxy
✗ from raft_dask import DaskClusterProxy

Initialize a Dask cluster with RAFT acceleration. Requires CUDA 12.x and a running cluster backend (e.g., UCX).

from raft_dask.cluster import DaskClusterProxy cluster = DaskClusterProxy() print(cluster) # Use with Dask client import dask.distributed as dd client = dd.Client(cluster) print(client)
Debug
Known issues
breakingRequires CUDA 12.2+ as of v25.12.00; CUDA 11 support dropped entirely.
fix
Upgrade to CUDA 12.2 or later, or use raft-dask-cu12=25.10.00 for CUDA 11.
affects: >=25.12.00
breakingImport paths changed: most top-level imports now come from 'raft_dask' subpackages. Old 'raft.dask.*' patterns removed in v25.12.
fix
Replace 'import raft.dask' with 'import raft_dask' and adjust imports accordingly.
affects: >=25.12.00
gotchaDaskClusterProxy assumes UCX-Py is installed and configured for GPU communication; missing UCX leads to runtime errors.
fix
Install ucx-py and configure UCX transport: pip install ucx-py.
affects: all
Upgrade
Version history
26.6.0latest on PyPI · released Jun 9, 2026
Audit
Dependencies

No dependency data recorded yet.

Agent activity
22 hits · last 30 days
node
18
OpenAI (training)
1
Resources
githubhttps://github.com/rapidsai/raft ↗
packagehttps://pypi.org/project/raft-dask-cu12/ ↗
homepagehttps://raft.rapids.ai ↗
docshttps://docs.rapids.ai/api/raft/stable/ ↗
raft-dask-cu12 — pip install raft-dask-cu12 · libregistry