Approximate kNN

[Kaixhin]

The original works use approximate kNN methods - specifically a k-d tree - to make inference scalable. This should be done, either by writing a custom k-d tree that supports online updates to the index, or modifying existing approximate methods from libraries such as scikit-learn, FAISS or PyNNDescent.