search

teddykoker / torchsort

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

Add cuda 12.1, PyTorch 2.1 releases #79

Closed teddykoker closed 8 months ago

teddykoker commented 8 months ago

Add binaries for cuda 12.1, pytorch 2.1 (#78)