topk
High-performance KNN similarity functions in Python, optimized for sparse matrices
Pytorch implementation of a differentiable topk function, a balanced and imbalanced top-k loss for deep learning
