torchsort (pypi): Differentiable sorting and ranking in PyTorch
Keywords: cuda-kernel|pytorch|ranking|sort