Fast Differentiable Sorting and Ranking
☆616Feb 15, 2024Updated 2 years ago
Alternatives and similar repositories for fast-soft-sort
Users that are interested in fast-soft-sort are comparing it to the libraries listed below
Sorting:
- Fast, differentiable sorting and ranking in PyTorch☆850Jun 9, 2025Updated 8 months ago
- Differentiable Sorting Networks☆125Aug 9, 2023Updated 2 years ago
- ☆213Jan 31, 2022Updated 4 years ago
- Differentiable bitonic sorting☆141Jul 10, 2020Updated 5 years ago
- Pytorch library for fast transformer implementations☆1,763Mar 23, 2023Updated 2 years ago
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,123Apr 20, 2022Updated 3 years ago
- ☆50Oct 22, 2020Updated 5 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Jul 22, 2020Updated 5 years ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,627Mar 25, 2022Updated 3 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,159Feb 6, 2026Updated 3 weeks ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆259Oct 29, 2023Updated 2 years ago
- Differentiable convex optimization layers☆2,063Updated this week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,415Feb 20, 2026Updated last week
- Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…☆348Dec 21, 2021Updated 4 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,163Mar 22, 2024Updated last year
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,027Dec 17, 2025Updated 2 months ago
- Hopfield Networks is All You Need☆1,900Apr 23, 2023Updated 2 years ago
- Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses☆194Sep 19, 2023Updated 2 years ago
- ☆774Jan 27, 2024Updated 2 years ago
- Differentiable Top-k Classification Learning☆92Jan 3, 2023Updated 3 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Aug 21, 2025Updated 6 months ago
- A GPipe implementation in PyTorch☆863Jul 25, 2024Updated last year
- Documentation:☆129May 22, 2023Updated 2 years ago
- A framework for implementing equivariant DL☆10May 25, 2021Updated 4 years ago
- A highly efficient implementation of Gaussian Processes in PyTorch☆3,835Updated this week
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆91Oct 19, 2022Updated 3 years ago
- PyTorch implementation of L2L execution algorithm☆108Jan 16, 2023Updated 3 years ago
- JAX-based neural network library☆3,193Updated this week
- PiRank: Learning to Rank via Differentiable Sorting☆61Mar 24, 2023Updated 2 years ago
- Manifold-learning flows (ℳ-flows)☆233Nov 13, 2020Updated 5 years ago
- Structured state space sequence models☆2,854Jul 17, 2024Updated last year
- Constrained optimization toolkit for PyTorch☆706Jul 29, 2025Updated 7 months ago
- maximal update parametrization (µP)☆1,686Jul 17, 2024Updated last year
- Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.☆6,350Apr 4, 2025Updated 11 months ago
- A Python toolbox for performing gradient-free optimization☆4,152May 30, 2025Updated 9 months ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,548Jul 31, 2021Updated 4 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,377Mar 1, 2024Updated 2 years ago
- The entmax mapping and its loss, a family of sparse softmax alternatives.☆461Jun 22, 2024Updated last year
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Feb 21, 2020Updated 6 years ago