rusty1s / pytorch_unique
PyTorch Extension Library of Optimized Unique Operation
☆37Updated 6 years ago
Alternatives and similar repositories for pytorch_unique:
Users that are interested in pytorch_unique are comparing it to the libraries listed below
- [ICLR 2019] Learning Representations of Sets through Optimized Permutations☆36Updated 5 years ago
- Implementation of Adversarial Variational Optimization in PyTorch☆43Updated 6 years ago
- Compare outputs between layers written in Tensorflow and layers written in Pytorch☆72Updated 6 years ago
- (Batched) advanced indexing for PyTorch.☆53Updated 2 months ago
- Implements an infinite sum of poisson-weighted convolutions☆26Updated 6 years ago
- TensorFlow, PyTorch and Numpy layers for generating Orthogonal Polynomials☆31Updated 6 years ago
- PyTorch Impl. of Prediction Optimizer (to stabilize GAN training)☆31Updated 6 years ago
- Code for the paper "Learning sparse transformations through backpropagation"☆43Updated 5 years ago
- Code for SegTree Transformer (ICLR-RLGM 2019).☆27Updated 5 years ago
- tunz's CUDA pytorch operator (MaskedSoftmax)☆75Updated 5 years ago
- A PyTorch library for differentiable submodular minimization.☆58Updated 7 years ago
- SparseMAP: differentiable sparse structure inference☆111Updated 6 years ago
- Learning and Reasoning with Graph-Structured Data (ICML 2019 Workshop)☆26Updated 5 years ago
- An implementation of shampoo☆74Updated 6 years ago
- Reference implementation for Structured Prediction with Deep Value Networks☆54Updated 7 years ago
- ☆74Updated 7 years ago
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- Open source implementation of SeaRNN (ICLR 2018, https://openreview.net/forum?id=HkUR_y-RZ)☆48Updated 6 years ago
- ☆12Updated 7 years ago
- Differentiable bitonic sorting☆138Updated 4 years ago
- Code for "Aggregated Momentum: Stability Through Passive Damping", Lucas et al. 2018☆34Updated 6 years ago
- a replicate of https://arxiv.org/pdf/1711.00937.pdf☆16Updated 7 years ago
- The Limited Multi-Label Projection Layer☆58Updated 7 months ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆118Updated last month
- ☆28Updated 4 years ago
- The Variational Homoencoder: Learning to learn high capacity generative models from few examples☆34Updated last year
- Implementation of AlphaZero in PyTorch.☆10Updated 5 years ago
- Code for "Deep Convolutional Networks as shallow Gaussian Processes"☆39Updated 5 years ago
- Re-implementation of the Noise Contrastive Estimation algorithm for pyTorch, following "Noise-contrastive estimation: A new estimation pr…☆45Updated 5 years ago
- Odds and Ends and Things I've implemented.☆78Updated 6 years ago