GitHub repository: 5had3z/torch-discounted-cumsum-nd
Primary language: Cuda
Description: PyTorch Discounted Cumsum with Autograd (CPU + CUDA)