arrigonialberto86 / set_transformer
A TensorFlow implementation of the paper 'Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks'
☆31Updated 11 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
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 3 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆22Updated 5 years ago
- ☆16Updated 3 years ago
- Code to accompany the paper Radial Bayesian Neural Networks: Beyond Discrete Support In Large-Scale Bayesian Deep Learning☆33Updated 5 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
- ☆49Updated 4 years ago
- Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation☆69Updated 4 years ago
- ☆13Updated 9 months ago
- Code accompanying paper "Set Norm and Equivariant Skip Connections: Putting the Deep in Deep Sets."☆25Updated 2 years ago
- Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"☆79Updated 3 years ago
- ☆36Updated 2 years ago
- Implementation of the Functional Neural Process models☆43Updated 4 years ago
- Reliable Uncertainty Estimates in Deep Neural Networks using Noise Contrastive Priors☆62Updated 5 years ago
- Variational Autoencoders & Normalizing Flows Project☆18Updated 8 years ago
- Reproduction work of "Neural Relational Inference for Interacting Systems" in Chainer☆33Updated 6 years ago
- Code for "Training Deep Energy-Based Models with f-Divergence Minimization" ICML 2020☆36Updated 2 years ago
- ☆24Updated 5 years ago
- Uncertainty in Conditional Average Treatment Effect Estimation☆31Updated 4 years ago
- Contains code relating to this arxiv paper https://arxiv.org/abs/1802.03761☆37Updated 7 years ago
- Codebase for Learning Invariances in Neural Networks☆95Updated 2 years ago
- ☆29Updated 3 years ago
- Notes from NeurIPS 2019☆29Updated 5 years ago
- Pytorch implementation of VAEs for heterogeneous likelihoods.☆42Updated 2 years ago
- Autoregressive Energy Machines☆77Updated 2 years ago
- Implementation of the Neural Clustering Process algorithm in Pytorch☆31Updated 4 years ago
- "Oblique Decision Trees from Derivatives of ReLU Networks" (ICLR 2020, previously called "Locally Constant Networks")☆22Updated 3 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