rragundez / chunkdot

Multi-threaded matrix multiplication and cosine similarity calculations for dense and sparse matrices. Appropriate for calculating the K most similar items for a large number of items by chunking the item matrix representation (embeddings) and using Numba to accelerate the calculations.
78Updated 3 weeks ago

Alternatives and similar repositories for chunkdot:

Users that are interested in chunkdot are comparing it to the libraries listed below