juho-lee / set_transformer
Pytorch implementation of set transformer
☆563Updated 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
- ☆288Updated 2 years ago
- Pytorch implementation of Hyperspherical Variational Auto-Encoders☆358Updated 4 years ago
- Riemannian Adaptive Optimization Methods with pytorch optim☆894Updated 9 months ago
- Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"☆420Updated 2 years ago
- PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations☆1,036Updated last month
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,120Updated last year
- Geometric loss functions between point clouds, images and volumes☆613Updated last year
- Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…☆342Updated 3 years ago
- PyTorch implementation of a Variational Autoencoder with Gumbel-Softmax Distribution☆206Updated 6 years ago
- Deep InfoMax (DIM), or "Learning Deep Representations by Mutual Information Estimation and Maximization"☆806Updated 5 years ago
- Implementation of Gaussian Mixture Variational Autoencoder (GMVAE) for Unsupervised Clustering☆332Updated 4 years ago
- ☆188Updated 2 years ago
- Approximating Wasserstein distances with PyTorch☆457Updated last year
- Toolbox to integrate optimal transport loss functions using automatic differentiation and Sinkhorn's algorithm☆438Updated 6 years ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,607Updated 2 years ago
- Gaussian mixture models in PyTorch.