giannisdaras / smyrf
[NeurIPS 2020] Official Implementation: "SMYRF: Efficient Attention using Asymmetric Clustering".
☆47Updated last year
Related projects: ⓘ
- The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…☆66Updated last year
- Code for reversible recurrent neural networks☆38Updated 5 years ago
- ☆12Updated this week
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆97Updated 3 years ago
- ☆42Updated 3 years ago
- MTAdam: Automatic Balancing of Multiple Training Loss Terms☆34Updated 3 years ago
- Pytorch Implemetation for our NAACL2019 Paper "Riemannian Normalizing Flow on Variational Wasserstein Autoencoder for Text Modeling" http…☆62Updated 4 years ago
- Humans understand novel sentences by composing meanings and roles of core language components. In contrast, neural network models for nat…☆27Updated 4 years ago
- Code for the paper PermuteFormer☆43Updated 2 years ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"☆23Updated last year
- A python library for highly configurable transformers - easing model architecture search and experimentation.☆50Updated 2 years ago
- Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation☆69Updated 3 years ago
- A variant of Transformer-XL where the memory is updated not with a queue, but with attention☆45Updated 4 years ago
- GPT, but made only out of MLPs☆86Updated 3 years ago
- ☆24Updated 4 months ago
- ICML 2020, Estimating Generalization under Distribution Shifts via Domain-Invariant Representations☆21Updated 4 years ago
- LP-SparseMAP: Differentiable sparse structured prediction in coarse factor graphs☆41Updated 10 months ago
- ☆63Updated 2 years ago
- Code for "Rissanen Data Analysis: Examining Dataset Characteristics via Description Length" by Ethan Perez, Douwe Kiela, and Kyungyhun Ch…☆35Updated 3 years ago
- A GPT, made only of MLPs, in Jax☆55Updated 3 years ago
- Skyformer: Remodel Self-Attention with Gaussian Kernel and Nystr\"om Method (NeurIPS 2021)☆58Updated 2 years ago
- Axial Positional Embedding for Pytorch☆61Updated 3 years ago
- An implementation of Transformer with Expire-Span, a circuit for learning which memories to retain☆33Updated 3 years ago
- A Pytorch implementation of the optimal transport kernel embedding☆111Updated 3 years ago
- Code publication to the paper "Normalized Attention Without Probability Cage"☆16Updated 2 years ago
- This repository contains the code for running the character-level Sandwich Transformers from our ACL 2020 paper on Improving Transformer …☆55Updated 3 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆79Updated 2 years ago
- ☆64Updated 4 years ago
- ☆46Updated 3 years ago