lucidrains / isab-pytorch
An implementation of (Induced) Set Attention Block, from the Set Transformers paper
☆56Updated 2 years ago
Alternatives and similar repositories for isab-pytorch:
Users that are interested in isab-pytorch are comparing it to the libraries listed below
- ☆47Updated 2 years ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"☆25Updated last year
- Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation☆69Updated 4 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆50Updated 2 years ago
- ☆32Updated 5 years ago
- Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"☆78Updated 2 years ago
- A simple implementation of a deep linear Pytorch module☆19Updated 4 years ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- Graph neural network message passing reframed as a Transformer with local attention☆67Updated 2 years ago
- Code for the paper PermuteFormer☆42Updated 3 years ago
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆22Updated 2 years ago
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆88Updated 4 years ago
- Implementation of the Remixer Block from the Remixer paper, in Pytorch☆35Updated 3 years ago
- Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)☆48Updated last year
- General Invertible Transformations for Flow-based Generative Models☆17Updated 4 years ago
- ☆41Updated last year
- ☆21Updated last year
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆53Updated 2 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- A GPT, made only of MLPs, in Jax☆57Updated 3 years ago
- ☆33Updated last year
- Implementation of LogAvgExp for Pytorch☆33Updated 2 years ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Code for "SAM as an Optimal Relaxation of Bayes", ICLR 2023.☆25Updated last year
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated last year
- ☆37Updated 2 years ago
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆24Updated 2 years ago
- Code base for SRSGD.☆28Updated 4 years ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆25Updated 3 years ago
- [ICML'21 Oral] Improving Lossless Compression Rates via Monte Carlo Bits-Back Coding☆14Updated 3 years ago