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
- (Batched) advanced indexing for PyTorch.☆53Updated 3 months ago
- Implements an infinite sum of poisson-weighted convolutions☆26Updated 6 years ago
- Implementation of Adversarial Variational Optimization in PyTorch☆43Updated 6 years ago
- Code for SegTree Transformer (ICLR-RLGM 2019).☆27Updated 5 years ago
- Code for the paper "Learning sparse transformations through backpropagation"☆43Updated 5 years ago
- Compare outputs between layers written in Tensorflow and layers written in Pytorch☆72Updated 6 years ago
- tunz's CUDA pytorch operator (MaskedSoftmax)☆75Updated 6 years ago
- The Limited Multi-Label Projection Layer☆58Updated 9 months ago
- a replicate of https://arxiv.org/pdf/1711.00937.pdf☆16Updated 7 years ago
- Code for "Aggregated Momentum: Stability Through Passive Damping", Lucas et al. 2018☆34Updated 6 years ago
- PyTorch Impl. of Prediction Optimizer (to stabilize GAN training)☆31Updated 6 years ago
- A PyTorch library for differentiable submodular minimization.☆58Updated 7 years ago
- A PyTorch implementation of Conditional PixelCNNs☆27Updated 7 years ago
- Partially Adaptive Momentum Estimation method in the paper "Closing the Generalization Gap of Adaptive Gradient Methods in Training Deep …☆39Updated 2 years ago
- Linear-chain LSTM-CRFs and Convolutional CRFs in PyTorch.☆19Updated 7 years ago
- Odds and Ends and Things I've implemented.☆78Updated 6 years ago
- learning to search in pytorch☆110Updated 5 years ago
- The Variational Homoencoder: Learning to learn high capacity generative models from few examples☆34Updated last year
- A demo project for a computation graph implementation in C++.☆11Updated 5 years ago
- Implementation of AlphaZero in PyTorch.☆10Updated 6 years ago
- Re-implementation of the Noise Contrastive Estimation algorithm for pyTorch, following "Noise-contrastive estimation: A new estimation pr…☆45Updated 5 years ago
- Learning and Reasoning with Graph-Structured Data (ICML 2019 Workshop)☆26Updated 5 years ago
- An implementation of shampoo☆74Updated 7 years ago
- PyTorch implementation of CVPR'18 - Perturbative Neural Networks http://xujuefei.com/pnn.html☆57Updated 6 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆118Updated 3 months ago
- Open source implementation of SeaRNN (ICLR 2018, https://openreview.net/forum?id=HkUR_y-RZ)☆48Updated 6 years ago
- ReGAN: Sequence GAN using RE[INFORCE|LAX|BAR] based PG estimators☆40Updated 6 years ago
- TensorFlow, PyTorch and Numpy layers for generating Orthogonal Polynomials☆31Updated 6 years ago
- ☆28Updated 4 years ago