[FEA] Pre-filtered brute-force KNN

[cjnolet]

This is an important building block and can be implemented easily once we have a masked matmul primitive available. We can likely use the SDDMM primitive from cuSparse for the dot-product based distances and then eventually build our own for the non-dot-product based.

cc @benfred