Numpy / Numba implementation of IVF/PQ ANN index which is as fast as faiss

Follow the full discussion on Reddit.
Hi, just sharing with my recent github project fast-ivf in which I implemented from scratch IVF (+PQ) index using purely numpy/numba libraries. I did it mostly for an educational purpose. I also implemented something which I called `CompressedFastIVF` index which trains auto shallow autoencoder using kmeans assignments to reduce the dimensionality of the source embeddings, which seems to be a nice alternative for PQ method, at least on my data.

Visit Website

0

Comments

There's unfortunately not much to read here yet...

Discover the Best of Machine Learning.

Ever having issues keeping up with everything that's going on in Machine Learning? That's where we help. We're sending out a weekly digest, highlighting the Best of Machine Learning.

Join over 900 Machine Learning Engineers receiving our weekly digest.

Best of Machine LearningBest of Machine Learning

Discover the best guides, books, papers and news in Machine Learning, once per week.

Twitter

Best of ML
Home
RemoteML
Remote Jobs
Machine Learning Chat
Social Media
Team
(Esc)

Join 800 others in curating the best parts of Machine Learning!

Just one more step. We use social logins to make sure that real humans are curating the list!

or contribute on GitHub directly.