Differentiable sorting and ranking in PyTorch
Project description
Torchsort
Pure PyTorch implementation of Fast Differentiable Sorting and Ranking (Blondel et al.). Much of the code is copied from the original Numpy implementation at google-research/fast-soft-sort, with the Isotonic regression solver rewritten as a PyTorch C++ Extension.
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
torchsort-0.0.1.tar.gz
(5.4 kB
view hashes)
