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
777 stars 194 forks source link

[FEA] nn-descent Python API #1973

Open cjnolet opened 1 year ago

cjnolet commented 1 year ago

As the name suggests, nn-descent is immensely useful on its own, especially for computing all-neighbors graphs. It has its own c++ API but we need a Python API for it.

Intron7 commented 1 year ago

That would be amazing