rusty1s / pytorch_unique
PyTorch Extension Library of Optimized Unique Operation
☆37Updated 5 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
- Code for SegTree Transformer (ICLR-RLGM 2019).☆27Updated 5 years ago
- Implementation of Adversarial Variational Optimization in PyTorch☆43Updated 6 years ago
- (Batched) advanced indexing for PyTorch.☆53Updated last month
- Implements an infinite sum of poisson-weighted convolutions☆26Updated 6 years ago
- Learning and Reasoning with Graph-Structured Data (ICML 2019 Workshop)☆26Updated 5 years ago
- Code for the paper "Learning sparse transformations through backpropagation"☆43Updated 5 years ago
- SparseMAP: differentiable sparse structure inference☆111Updated 5 years ago
- tunz's CUDA pytorch operator (MaskedSoftmax)☆75Updated 5 years ago
- a replicate of https://arxiv.org/pdf/1711.00937.pdf☆16Updated 7 years ago
- ReGAN: Sequence GAN using RE[INFORCE|LAX|BAR] based PG estimators☆40Updated 6 years ago
- The Variational Homoencoder: Learning to learn high capacity generative models from few examples☆34Updated last year
- TensorFlow, PyTorch and Numpy layers for generating Orthogonal Polynomials☆31Updated 6 years ago
- Compare outputs between layers written in Tensorflow and layers written in Pytorch☆72Updated 6 years ago
- PyTorch Impl. of Prediction Optimizer (to stabilize GAN training)☆31Updated 6 years ago
- Generative Matching Networks☆30Updated 7 years ago
- A PyTorch implementation of Conditional PixelCNNs☆27Updated 7 years ago
- Differentiable bitonic sorting☆138Updated 4 years ago
- Computation using data flow graphs for scalable machine learning☆24Updated 6 years ago
- ☆74Updated 7 years ago
- The Limited Multi-Label Projection Layer☆58Updated 6 months ago
- Notes from NeurIPS 2019☆29Updated 5 years ago
- Open source implementation of SeaRNN (ICLR 2018, https://openreview.net/forum?id=HkUR_y-RZ)☆48Updated 6 years ago
- PyTorch C++ Extension Example☆15Updated 6 years ago
- General stats about NeurIPS 2019☆28Updated 4 years ago
- A public repository for our paper, Rao-Blackwellized Stochastic Gradients for Discrete Distributions☆22Updated 5 years ago
- A Toolkit for Training, Tracking, Saving Models and Syncing Results☆60Updated 4 years ago
- Weekly reading group on Graphs at Mila☆31Updated 5 years ago
- Notes from ICLR 2019☆24Updated 5 years ago
- ☆12Updated 7 years ago