kzl / universal-computation
Official codebase for Pretrained Transformers as Universal Computation Engines.
☆247Updated 3 years ago
Alternatives and similar repositories for universal-computation
Users that are interested in universal-computation are comparing it to the libraries listed below
Sorting:
- Trains Transformer model variants. Data isn't shuffled between batches.☆143Updated 2 years ago
- An attempt at the implementation of Glom, Geoffrey Hinton's new idea that integrates concepts from neural fields, top-down-bottom-up proc…☆193Updated 4 years ago
- Pytorch implementation of Compressive Transformers, from Deepmind☆157Updated 3 years ago
- Implementation of Feedback Transformer in Pytorch☆106Updated 4 years ago
- Fully featured implementation of Routing Transformer☆292Updated 3 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆81Updated 3 years ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆105Updated 3 years ago
- The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…☆67Updated 2 years ago
- Implementation of Memformer, a Memory-augmented Transformer, in Pytorch☆117Updated 4 years ago
- Contrastive Language-Image Pretraining☆142Updated 2 years ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆203Updated last year
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆127Updated 2 years ago
- Code for Multi-Head Attention: Collaborate Instead of Concatenate☆151Updated last year
- GPT, but made only out of MLPs☆89Updated 3 years ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆225Updated 3 years ago
- Code for the paper, "Distribution Augmentation for Generative Modeling", ICML 2020.☆123Updated 2 years ago
- A collection of code snippets for my PyTorch Lightning projects☆107Updated 4 years ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆100Updated 2 years ago
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆262Updated 3 years ago
- Code for "Recurrent Independent Mechanisms"☆118Updated 3 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆258Updated last year
- My implementation of DeepMind's Perceiver☆63Updated 4 years ago
- A library for evaluating representations.☆76Updated 3 years ago
- Differentiable Sorting Networks☆114Updated last year
- ☆376Updated last year
- ☆205Updated 2 years ago
- VQVAEs, GumbelSoftmaxes and friends☆560Updated 3 years ago
- Understanding the Difficulty of Training Transformers☆329Updated 2 years ago
- Bongard-LOGO is a Python code repository with the purpose of generating synthetic Bongard problems on a large scale with little human int…☆52Updated 3 years ago
- Understanding Training Dynamics of Deep ReLU Networks☆291Updated this week