Felix-Petersen / diffsort
Differentiable Sorting Networks
☆114Updated last year
Alternatives and similar repositories for diffsort:
Users that are interested in diffsort are comparing it to the libraries listed below
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆105Updated 3 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆115Updated last year
- ☆99Updated 3 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆258Updated last year
- Official Pytorch's implementation of RotoGrad☆76Updated last year
- Differentiable Top-k Classification Learning☆81Updated 2 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated last year
- The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…☆67Updated 2 years ago
- Reparameterize your PyTorch modules☆71Updated 4 years ago
- ☆133Updated 4 years ago
- Efficient reservoir sampling implementation for PyTorch☆106Updated 3 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆204Updated 8 months ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 2 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆106Updated 4 years ago
- Codebase for Learning Invariances in Neural Networks☆95Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆109Updated 2 years ago
- Simple Python scripts to clean up and flatten ArXiv LaTeX submissions.☆65Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆174Updated this week
- simple bibtex generator for any text with \cite{}☆31Updated 9 months ago
- Code to reproduce the results for Compositional Attention☆60Updated 2 years ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆129Updated 2 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 3 years ago
- A Machine Learning workflow for Slurm.☆149Updated 4 years ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆124Updated 9 months ago
- Meta Optimal Transport☆102Updated last year
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆62Updated 3 years ago
- Differentiable Optimizers with Perturbations in Pytorch☆68Updated 4 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆79Updated 3 years ago
- Structured matrices for compressing neural networks☆66Updated last year