GitHub repository: vispz/torchsort
Description: Fast, differentiable sorting and ranking in PyTorch