juho-lee / set_transformer
Pytorch implementation of set transformer
☆574Updated 5 years ago
Alternatives and similar repositories for set_transformer:
Users that are interested in set_transformer are comparing it to the libraries listed below
- ☆289Updated 2 years ago
- Pytorch implementation of Hyperspherical Variational Auto-Encoders☆361Updated 5 years ago
- Deep InfoMax (DIM), or "Learning Deep Representations by Mutual Information Estimation and Maximization"☆807Updated 5 years ago
- PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations☆1,051Updated 2 months ago
- Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"☆420Updated 2 years ago
- Efficient Graph Generation with Graph Recurrent Attention Networks, Deep Generative Model of Graphs, Graph Neural Networks, NeurIPS 2019☆477Updated last year
- Experiments for understanding disentanglement in VAE latent representations☆812Updated 2 years ago
- Geometric loss functions between point clouds, images and volumes☆621Updated last year
- Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…☆344Updated 3 years ago
- [NeurIPS 2019] Deep Set Prediction Networks☆100Updated 4 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,136Updated last year
- Riemannian Adaptive Optimization Methods with pytorch optim☆909Updated 11 months ago
- A curated list of awesome work on VAEs, disentanglement, representation learning, and generative models.☆805Updated 3 years ago
- An implementation of Performer, a linear attention-based transformer, in Pytorch☆1,120Updated 3 years ago
- PyTorch implementation of a Variational Autoencoder with Gumbel-Softmax Distribution☆208Updated 6 years ago
- ☆192Updated 2 years ago
- The entmax mapping and its loss, a family of sparse softmax alternatives.☆430Updated 9 months ago
- Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2…☆440Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models☆742Updated 2 years ago
- Minimalist implementation of VQ-VAE in Pytorch☆538Updated 3 years ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,610Updated 3 years ago
- Fast, differentiable sorting and ranking in PyTorch☆799Updated last year
- Approximating Wasserstein distances with PyTorch☆457Updated last year
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆260Updated 3 years ago
- Disentanglement library for PyTorch☆276Updated 2 years ago
- Implementation of E(n)-Equivariant Graph Neural Networks, in Pytorch☆455Updated 3 months ago
- PyTorch Extension Library of Optimized Scatter Operations☆1,626Updated 2 months ago
- Neural relational inference for interacting systems - pytorch☆753Updated 5 years ago
- Code for the paper: Putting An End to End-to-End: Gradient-Isolated Learning of Representations☆285Updated 2 years ago
- code for "Isolating Sources of Disentanglement in Variational Autoencoders".☆349Updated 2 years ago