rusty1s / pytorch_unique
PyTorch Extension Library of Optimized Unique Operation
☆37Updated 5 years ago
Related projects: ⓘ
- [ICLR 2019] Learning Representations of Sets through Optimized Permutations☆36Updated 5 years ago
- (Batched) advanced indexing for PyTorch.☆53Updated 8 months ago
- The Limited Multi-Label Projection Layer☆58Updated last month
- Code for the paper "Learning sparse transformations through backpropagation"☆42Updated 4 years ago
- Notes from NeurIPS 2019☆29Updated 4 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
- Generative Matching Networks☆31Updated 7 years ago
- ☆73Updated 7 years ago
- tunz's CUDA pytorch operator (MaskedSoftmax)☆74Updated 5 years ago
- Code for SegTree Transformer (ICLR-RLGM 2019).☆27Updated 4 years ago
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆87Updated 3 years ago
- Implements an infinite sum of poisson-weighted convolutions☆26Updated 6 years ago
- Partially Adaptive Momentum Estimation method in the paper "Closing the Generalization Gap of Adaptive Gradient Methods in Training Deep …☆39Updated last year
- The Variational Homoencoder: Learning to learn high capacity generative models from few examples☆33Updated last year
- Notes from ICLR 2019☆24Updated 5 years ago
- A pytorch implementation of Information Bottleneck GAN☆28Updated 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
- ReGAN: Sequence GAN using RE[INFORCE|LAX|BAR] based PG estimators☆41Updated 6 years ago
- ☆20Updated this week
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆53Updated 5 years ago
- TensorFlow, PyTorch and Numpy layers for generating Orthogonal Polynomials☆31Updated 6 years ago
- Code for "Deep Convolutional Networks as shallow Gaussian Processes"☆38Updated 5 years ago
- A PyTorch implementation of Conditional PixelCNNs☆27Updated 6 years ago
- Odds and Ends and Things I've implemented.☆78Updated 5 years ago
- Code for "Aggregated Momentum: Stability Through Passive Damping", Lucas et al. 2018☆34Updated 5 years ago
- PyTorch Impl. of Prediction Optimizer (to stabilize GAN training)☆31Updated 6 years ago
- Reference implementation for Structured Prediction with Deep Value Networks☆55Updated 7 years ago
- An implementation of shampoo☆73Updated 6 years ago