torchpsort

KO | EN

gitlite — search

#javascript #python #hacktoberfest #llm #golang #android #rust #deep-learning #docker #java #nodejs #windows #awesome-list #pytorch

(★ 10)

Fast, differentiable sorting and ranking without custom C++ or CUDA extensions

torchpsort 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

Fast, differentiable sorting and ranking **without custom C++ or CUDA extensions**. This is a lightweight implementation of [Fast Differentiable Sorting and Ranking (Blondel et al.)](https://arxiv.org/abs/2002.08871) based on [fast-soft-sort](https://github.com/google-research/fast-soft-sort) and inspired by [torchsort](https://github.com/teddykoker/torchsort). Unlike [torchsort](https://github.com/teddykoker/torchsort), this version requires no custom C++ or CUDA extensions, making it easy to install and portable across platforms. While optimized C++/CUDA implementations may have a performance advantage for extremely large sequence lengths, this version is optimized for typical deep learning workflows. Try it here:
[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/RektPunk/torchpsort/blob/main/examples/notebook.ipynb)

## Installation

```bash
pip install torchpsort
```

> [!NOTE]
> The CPU forward implementation uses [Numba](https://numba.pydata.org/) for JIT compilation like [fast-soft-sort](https://github.com/google-research/fast-soft-sort). Numba is installed automatically as a dependency of `torchpsort`. On CUDA tensors, the forward pass uses a separate chunked GPU implementation instead.

torchpsort

torchpsort 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

torchpsort

torchpsort 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README