rapidsai / raft

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.
https://docs.rapids.ai/api/raft/stable/
Apache License 2.0
747 stars 189 forks source link

[REVIEW] Adjust UCX dependencies #2304

Closed pentschev closed 4 months ago

pentschev commented 4 months ago

Add distributed-ucxx as a required dependency to raft-dask and remove direct dependencies to ucx/ucx-proc in favor of transitive installation from ucx-py/distributed-ucxx.

cjnolet commented 4 months ago

/merge