juho-lee / set_transformer
Pytorch implementation of set transformer
☆552Updated 4 years ago
Alternatives and similar repositories for set_transformer:
Users that are interested in set_transformer are comparing it to the libraries listed below
- ☆284Updated last year
- Geometric loss functions between point clouds, images and volumes☆606Updated 10 months ago
- Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…☆340Updated 2 years ago
- PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations☆1,021Updated last month
- Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"☆418Updated 2 years ago
- PyTorch Extension Library of Optimized Graph Cluster Algorithms☆838Updated last month
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆255Updated 3 years ago
- PyTorch Extension Library of Optimized Scatter Operations☆1,576Updated 2 weeks ago
- Experiments for understanding disentanglement in VAE latent representations☆801Updated last year
- [NeurIPS'19] Deep Equilibrium Models☆732Updated 2 years ago
- [NeurIPS'22] Tokenized Graph Transformer (TokenGT), in PyTorch☆322Updated last year
- Implementation of Principal Neighbourhood Aggregation for Graph Neural Networks in PyTorch, DGL and PyTorch Geometric☆343Updated 2 years ago
- Deep InfoMax (DIM), or "Learning Deep Representations by Mutual Information Estimation and Maximization"☆807Updated 5 years ago
- Riemannian Adaptive Optimization Methods with pytorch optim☆862Updated 7 months ago
- Approximating Wasserstein distances with PyTorch☆456Updated last year
- Fast Differentiable Sorting and Ranking☆574Updated 10 months ago
- Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2…☆437Updated 2 years ago
- A curated list of awesome work on VAEs, disentanglement, representation learning, and generative models.☆791Updated 3 years ago
- PyTorch implementation of a Variational Autoencoder with Gumbel-Softmax Distribution☆204Updated 6 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,104Updated last year
- Efficient Graph Generation with Graph Recurrent Attention Networks, Deep Generative Model of Graphs, Graph Neural Networks, NeurIPS 2019☆473Updated last year
- Pytorch implementation of Hyperspherical Variational Auto-Encoders☆353Updated 4 years ago
- Toolbox to integrate optimal transport loss functions using automatic differentiation and Sinkhorn's algorithm☆435Updated 6 years ago
- Contrastive Learning of Structured World Models☆388Updated 4 years ago
- [NeurIPS 2019] Deep Set Prediction Networks☆100Updated 4 years ago
- Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"☆284Updated 3 years ago
- An implementation of Performer, a linear attention-based transformer, in Pytorch☆1,102Updated 2 years ago
- Code for the paper: Putting An End to End-to-End: Gradient-Isolated Learning of Representations☆285Updated last year
- Disentanglement library for PyTorch☆275Updated 2 years ago
- Implementation of E(n)-Equivariant Graph Neural Networks, in Pytorch☆434Updated last week