Fast, differentiable sorting and ranking in PyTorch
☆865Mar 3, 2026Updated 7 months ago
Alternatives and similar repositories for torchsort
Users that are interested in torchsort are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Fast Differentiable Sorting and Ranking☆621Feb 15, 2024Updated 2 years ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,485May 2, 2025Updated last year
- functorch is JAX-like composable function transforms for PyTorch.☆1,433Aug 21, 2025Updated last year
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,194Jul 17, 2026Updated 3 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,630Aug 26, 2026Updated 2 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- torch-optimizer -- collection of optimizers for Pytorch☆3,172Mar 22, 2024Updated 2 years ago
- Differentiable Sorting Networks☆130Aug 9, 2023Updated 3 years ago
- Differentiable convex optimization layers☆2,149Oct 24, 2026Updated last week
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,624Mar 25, 2022Updated 4 years ago
- Pytorch library for fast transformer implementations☆1,785Mar 23, 2023Updated 3 years ago
- Library for 8-bit optimizers and quantization routines.☆777Aug 18, 2022Updated 4 years ago
- PyTorch Extension Library of Optimized Scatter Operations☆1,749Jun 3, 2026Updated 5 months ago
- Constrained optimization toolkit for PyTorch☆712Jun 7, 2026Updated 4 months ago
- Differentiable bitonic sorting☆141Jul 10, 2020Updated 6 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,134Apr 20, 2022Updated 4 years ago
- The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).☆567Jul 13, 2023Updated 3 years ago
- High performance model preprocessing library on PyTorch☆639Mar 29, 2024Updated 2 years ago
- 🐍 Geometric Computer Vision Library for Spatial AI☆12,128Updated this week
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆681Feb 21, 2020Updated 6 years ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,995Jun 16, 2024Updated 2 years ago
- Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses☆196Sep 19, 2023Updated 3 years ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆310Apr 15, 2026Updated 6 months ago
- A Python toolbox for performing gradient-free optimization☆4,221Oct 20, 2026Updated 2 weeks ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆260Oct 29, 2023Updated 3 years ago
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,344Updated this week
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,265Sep 16, 2026Updated last month
- Code for "Stochastic Optimization of Sorting Networks using Continuous Relaxations", ICLR 2019.☆151Mar 24, 2023Updated 3 years ago
- MADGRAD Optimization Method☆801Jan 27, 2025Updated last year
- Reformer, the efficient Transformer, in Pytorch☆2,195Jun 21, 2023Updated 3 years ago
- Model interpretability and understanding for PyTorch☆5,719Updated this week
- On the Variance of the Adaptive Learning Rate and Beyond☆2,548Jul 31, 2021Updated 5 years ago
- Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains☆1,725Oct 8, 2023Updated 3 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆31,439Updated this week
- Profiling and inspecting memory in pytorch☆1,079Aug 12, 2026Updated 2 months ago
- PyTorch implementation of L2L execution algorithm☆109Jan 16, 2023Updated 3 years ago
- Repository for NeurIPS 2020 Spotlight "AdaBelief Optimizer: Adapting stepsizes by the belief in observed gradients"☆1,069Aug 9, 2024Updated 2 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆9,002Oct 26, 2026Updated last week
- allRank is a framework for training learning-to-rank neural models based on PyTorch.☆1,013Aug 6, 2024Updated 2 years ago
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,753Jan 20, 2026Updated 9 months ago