search

thomasahle / tinyknn

A tiny approximate K-Nearest Neighbour library in Python based on Fast Product Quantization and IVF
GNU Affero General Public License v3.0
15 stars 3 forks source link

Add numba #7

Open thomasahle opened 1 year ago

thomasahle commented 1 year ago

PyNNDescent (https://github.com/lmcinnes/pynndescent) is able to get a lot of speed (presumably) by using numba. This should be relatively easy to add to fast_pq as well.

thomasahle commented 1 year ago

This seems harder than expected. A lot of the numpy functions we use, like argpartition, aren't even supported by numba.