johnhw / differentiable_sorting
Differentiable bitonic sorting
☆138Updated 4 years ago
Alternatives and similar repositories for differentiable_sorting:
Users that are interested in differentiable_sorting are comparing it to the libraries listed below
- Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses☆183Updated last year
- Basic experiment framework for tensorflow.☆92Updated 3 years ago
- This repository is no longer maintained. Check☆81Updated 4 years ago
- Loss Patterns of Neural Networks☆83Updated 3 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- Autoregressive Energy Machines☆77Updated 2 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- Optimizing control variates for black-box gradient estimation☆162Updated 5 years ago
- Odds and Ends and Things I've implemented.☆78Updated 6 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 7 months ago
- learning to search in pytorch☆110Updated 4 years ago
- a light weight experiment reproducibility toolset☆40Updated 4 years ago
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆53Updated 5 years ago
- Implements pytorch code for the Accelerated SGD algorithm.☆215Updated 6 years ago
- A PyTorch implementation of Conditional PixelCNNs☆27Updated 7 years ago
- Code in PyTorch for the convex combination linear IAF and the Householder Flow, J.M. Tomczak & M. Welling☆91Updated 7 years ago
- Efficient reservoir sampling implementation for PyTorch☆107Updated 3 years ago
- Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.☆106Updated 2 years ago
- Pytorch Custom CUDA kernel for searchsorted☆135Updated last year
- Hypergradient descent☆145Updated 8 months ago
- Reparameterize your PyTorch modules☆70Updated 4 years ago
- Understanding normalizing flows☆131Updated 5 years ago
- [NeurIPS 2019] Deep Set Prediction Networks☆100Updated 4 years ago
- SparseMAP: differentiable sparse structure inference☆111Updated 6 years ago
- Forward-mode Automatic Differentiation for TensorFlow☆140Updated 6 years ago
- Implementation of REBAR in PyTorch☆17Updated 6 years ago
- Structured matrices for compressing neural networks☆66Updated last year
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 6 years ago
- Memory efficient MAML using gradient checkpointing☆83Updated 5 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆118Updated last month