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.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

rragundez/chunkdot Stargazers