Differentiable parallel approximate sorting networks

Simple (approximate) differentiable sorting and argsorting of vectors, so you can backprop through a sort() or argmax(). Using softmax sorting networks

For more details, visit the source.