This isn't much, and that's okay. It's just an easy little shim for building indices in a GPU-agnostic fashion.
I also included L2 normalization, as recommended by FAISS docs. I've personally experienced the poor performance of non-normalized inputs, so I'm providing it by default.
This isn't much, and that's okay. It's just an easy little shim for building indices in a GPU-agnostic fashion.
I also included L2 normalization, as recommended by FAISS docs. I've personally experienced the poor performance of non-normalized inputs, so I'm providing it by default.