Fast, differentiable sorting and ranking in PyTorch
-
Updated
Jun 9, 2025 - Python
Fast, differentiable sorting and ranking in PyTorch
a custom CUDA kernel for windowed matrix multiplication
Add a description, image, and links to the cuda-kernel topic page so that developers can more easily learn about it.
To associate your repository with the cuda-kernel topic, visit your repo's landing page and select "manage topics."