juho-lee / set_transformerView external linksLinks
Pytorch implementation of set transformer
☆649Feb 11, 2020Updated 6 years ago
Alternatives and similar repositories for set_transformer
Users that are interested in set_transformer are comparing it to the libraries listed below
Sorting:
- A TensorFlow implementation of the paper 'Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks'☆31May 3, 2024Updated last year
- A neural network architecture for prediction on sets☆24May 30, 2022Updated 3 years ago
- ☆311Jan 21, 2023Updated 3 years ago
- Pytorch implementation of neural processes and variants☆29Jul 25, 2024Updated last year
- Code accompanying paper "Set Norm and Equivariant Skip Connections: Putting the Deep in Deep Sets."☆30Oct 11, 2022Updated 3 years ago
- Pytorch Implementation of the Stacked Capsule Autoencoders☆108Oct 16, 2020Updated 5 years ago
- [NeurIPS 2019] Deep Set Prediction Networks☆101Sep 18, 2020Updated 5 years ago
- PyTorch re-implementation of parts of "Deep Sets" (NIPS 2017)☆73May 3, 2018Updated 7 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,194Aug 22, 2023Updated 2 years ago
- [CVPR'21] SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Data, in PyTorch☆78Oct 20, 2022Updated 3 years ago
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆67Jan 10, 2023Updated 3 years ago
- Pytorch implementation of Neural Processes for functions and images☆236Feb 8, 2022Updated 4 years ago
- Pytorch implementation of “MetaPerturb: Transferable Regularizer for Heterogeneous Tasks and Architectures” (NeurIPS 2020 spotlight)☆13Jul 22, 2021Updated 4 years ago
- Code accompanying the ICML-2018 paper "Gradient-Based Meta-Learning with Learned Layerwise Metric and Subspace"☆38Jan 18, 2019Updated 7 years ago
- Implementation of Slot Attention from GoogleAI☆480Aug 20, 2024Updated last year
- A collection of extensions and data-loaders for few-shot learning & meta-learning in PyTorch☆2,052Jul 17, 2023Updated 2 years ago
- The entmax mapping and its loss, a family of sparse softmax alternatives.☆459Jun 22, 2024Updated last year
- Graph Neural Network Library for PyTorch☆23,469Updated this week
- Official implementation of Transformer Neural Processes☆78Aug 21, 2022Updated 3 years ago
- This repository contains notebook implementations of the following Neural Process variants: Conditional Neural Processes (CNPs), Neural P…☆1,015Jan 19, 2021Updated 5 years ago
- Variational Reinforcement Learning☆17Jul 25, 2024Updated last year
- ☆15Sep 11, 2022Updated 3 years ago
- Neural likelihood-free methods in PyTorch.☆39Feb 11, 2020Updated 6 years ago
- Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"☆81Mar 4, 2022Updated 3 years ago
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆41Oct 4, 2023Updated 2 years ago
- Code for: "Fast and Flexible Multi-Task Classification Using Conditional Neural Adaptive Processes" and "TaskNorm: Rethinking Batch Norma…☆162Jan 14, 2021Updated 5 years ago
- SetToGraph paper repository☆22Sep 27, 2020Updated 5 years ago
- This is the official source code for Sequential Neural Processes.☆40Dec 8, 2022Updated 3 years ago
- ☆81Feb 10, 2022Updated 4 years ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,629Mar 25, 2022Updated 3 years ago
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆270Aug 10, 2021Updated 4 years ago
- ☆42Jan 22, 2024Updated 2 years ago
- Official Code Repository for the paper "Neural Mask Generator: Learning to Generate Adaptive Word Maskings for Language Model Adaptation …☆20Jun 19, 2023Updated 2 years ago
- Pytorch library for fast transformer implementations☆1,761Mar 23, 2023Updated 2 years ago
- Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.☆228Jun 12, 2024Updated last year
- Contrastive Learning of Structured World Models☆396Jun 3, 2020Updated 5 years ago
- A fully differentiable set autoencoder☆17Apr 3, 2024Updated last year
- Normalizing flows in PyTorch☆991Dec 27, 2024Updated last year
- GraphRNN: Generating Realistic Graphs with Deep Auto-regressive Models☆735Jan 20, 2022Updated 4 years ago