jaketae / fnet
PyTorch implementation of FNet: Mixing Tokens with Fourier transforms
☆25Updated 3 years ago
Alternatives and similar repositories for fnet:
Users that are interested in fnet are comparing it to the libraries listed below
- Implementation of the Remixer Block from the Remixer paper, in Pytorch☆35Updated 3 years ago
- A simple implementation of a deep linear Pytorch module☆19Updated 4 years ago
- A python library for highly configurable transformers - easing model architecture search and experimentation.☆49Updated 3 years ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Updated 2 years ago
- A Pytorch implementation of Attention on Attention module (both self and guided variants), for Visual Question Answering☆41Updated 4 years ago
- Implementation of Multistream Transformers in Pytorch☆53Updated 3 years ago
- Axial Positional Embedding for Pytorch☆69Updated last week
- Skyformer: Remodel Self-Attention with Gaussian Kernel and Nystr\"om Method (NeurIPS 2021)☆60Updated 2 years ago
- Local Attention - Flax module for Jax☆20Updated 3 years ago
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- Code for the paper "Query-Key Normalization for Transformers"☆37Updated 3 years ago
- ☆21Updated last year
- Unofficial PyTorch implementation of "Step-unrolled Denoising Autoencoders for Text Generation"☆23Updated 2 years ago
- A simple Transformer where the softmax has been replaced with normalization☆19Updated 4 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆53Updated 2 years ago
- An implementation of Transformer with Expire-Span, a circuit for learning which memories to retain☆33Updated 4 years ago
- A convolution-free, transformer-only version of the CycleGAN framework☆33Updated 3 years ago
- Unofficial PyTorch implementation of Google's FNet: Mixing Tokens with Fourier Transforms. With checkpoints.☆72Updated 2 years ago
- Project page for paper Self-supervised Representation Learning with Relative Predictive Coding☆17Updated 3 years ago
- PyTorch reimplementation of the paper "HyperMixer: An MLP-based Green AI Alternative to Transformers" [arXiv 2022].☆17Updated 2 years ago
- Code for the paper PermuteFormer☆42Updated 3 years ago
- An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols☆15Updated 3 years ago
- General Invertible Transformations for Flow-based Generative Models☆17Updated 4 years ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆48Updated 3 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
- Implementation of TableFormer, Robust Transformer Modeling for Table-Text Encoding, in Pytorch☆37Updated 2 years ago
- [NeurIPS 2020] Official Implementation: "SMYRF: Efficient Attention using Asymmetric Clustering".☆48Updated last year
- solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning☆23Updated 2 years ago
- Another attempt at a long-context / efficient transformer by me☆37Updated 2 years ago
- ☆29Updated 2 years ago