Hdbscan: HDBSCAN on GPU?

Created on 5 May 2017 · 7Comments · Source: scikit-learn-contrib/hdbscan

Hi guys,

Will there be any plan to develop a GPU enabled version of this amazing algorithm/package?

Unfortunately I have to admit that I'm new to GPU so won't be able to contribute anytime soon...

Thanks.

Source

esvhd

Most helpful comment

We are interested in doing this on RAPIDS cuML at some point. We're not sure when, but it's on our radar.

cjnolet on 2 May 2020

👍7 🎉3

All 7 comments

I'm not really a GPU programmer either, so it isn't on my immediate horizon. If I ever get around to writing an experimental version using numba instead of cython I will try to ensure that it can accept GPUs as a target architecture, but I really don't have any immediate plans to embark on that.

lmcinnes on 6 May 2017

Noted, thanks for getting back. Thought it was worth asking anyway. Perhaps others would be able to pick this up in the future.

esvhd on 6 May 2017

I would be happy to support any efforts -- I'm just not in a position to make them myself right now. Thanks for the suggestion though.

lmcinnes on 6 May 2017

@lmcinnes
I do some stuff with high level CNNs with Pytorch (best library for GPU + CNNs).
I found your library and education materials in the docs amazing.

Probably if there is some step / function in HDBSCAN pipeline that involves heavy matrix multiplication (to be honest - this is the best that CNN libraries can do and I am familiar with) - I could lend a hand in porting that to Pytorch.