arrigonialberto86 / set_transformer
A TensorFlow implementation of the paper 'Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks'
☆30Updated 8 months ago
Alternatives and similar repositories for set_transformer:
Users that are interested in set_transformer are comparing it to the libraries listed below
- ICML 2020 Paper: Latent Variable Modelling with Hyperbolic Normalizing Flows☆53Updated 2 years ago
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆42Updated last year
- ☆16Updated 3 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 2 years ago
- ☆13Updated 6 months ago
- ☆49Updated 4 years ago
- Code to accompany the paper Radial Bayesian Neural Networks: Beyond Discrete Support In Large-Scale Bayesian Deep Learning☆33Updated 4 years ago
- Code for "oi-VAE: Output Interpretable VAEs for Nonlinear Group Factor Analysis"☆26Updated 4 years ago
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆25Updated 3 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆21Updated 4 years ago
- Pretrained TorchVision models on CIFAR10 dataset (with weights)☆24Updated 4 years ago
- Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation☆69Updated 4 years ago
- Implementation of the Functional Neural Process models☆43Updated 4 years ago
- Reproducing the paper "Variational Sparse Coding" for the ICLR 2019 Reproducibility Challenge☆61Updated last year
- Codebase for Learning Invariances in Neural Networks☆93Updated 2 years ago
- Code for "Bridging the Gap between f-GANs and Wasserstein GANs", ICML 2020☆14Updated 4 years ago
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆56Updated 2 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 6 years ago
- Autoregressive Energy Machines☆77Updated 2 years ago
- ☆53Updated 6 months ago
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆53Updated 5 years ago
- ☆19Updated 2 years ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆25Updated 3 years ago
- ☆29Updated 3 years ago
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆88Updated 3 years ago
- Notes from NeurIPS 2019☆29Updated 5 years ago
- The original code for the paper "How to train your MAML" along with a replication of the original "Model Agnostic Meta Learning" (MAML) p…☆40Updated 4 years ago
- Pytorch implementation of VAEs for heterogeneous likelihoods.☆42Updated 2 years ago
- Uncertainty in Conditional Average Treatment Effect Estimation☆29Updated 3 years ago