GitHub repository: mgrankin/torchsearchsorted
Description: Pytorch Custom CUDA kernel for searchsorted