Just brute force your embeddings

When I wrote embedded C code in the 2000s we latched onto something Raymond Chen, wrote in passing

My O(n) algorithm can run circles around your O(log n) algorithm; why much of what you learned in school simply doesn’t matter

True of a sorting algorithm. True of vector search.

People think they need a vector database. But often they have like 1m docs to search. Numpy can brute-force an array of float32s very fast:

This is literally just this line of Python code, on 384 dim embeddings, running on my M4 MBP.

# Dot product against all
scores = self.doc_vectors @ query_vector.astype(np.float32, copy=False)

I work with a lot of teams that don’t need the complexity of a vector database. They have ~1m documents to search. They have low query traffic, and write their embeddings all up front. They don’t need to buy a multi-million dollar vector database, or spend 6 months learning to operate it.

For low enough n, just brute force the embeddings until you can’t bear to. As fellow search traveler Jo Kristian Bergum says “an exhaustive search may be all you need”.

A little past that, maybe think about a database? Or heck, just load them all in memory in FAISS or something and call it done.

As a footnote, this is just naive numpy operations, it could probably be faster. As Andreas Erickson says, throughput can be improved here by giving threads more than one query during the scan). And we could be doing a lot more during the search to collect into a top-n heap than what numpy does.