teddykoker / torchsort

Fast, differentiable sorting and ranking in PyTorch
https://pypi.org/project/torchsort/
Apache License 2.0
767 stars 33 forks source link

Cuda Kernel #8

Closed teddykoker closed 3 years ago

teddykoker commented 3 years ago

Implement isotonic regression forward and backward pass in CUDA. Close #2