kzl / universal-computation
Official codebase for Pretrained Transformers as Universal Computation Engines.
☆246Updated 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
- 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
- Trains Transformer model variants. Data isn't shuffled between batches.☆141Updated 2 years ago
- Pytorch implementation of Compressive Transformers, from Deepmind☆156Updated 3 years ago
- Implementation of Feedback Transformer in Pytorch☆105Updated 4 years ago
- Fully featured implementation of Routing Transformer☆291Updated 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
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆103Updated 3 years ago
- Differentiable Sorting Networks☆114Updated last year
- Understanding Training Dynamics of Deep ReLU Networks☆285Updated 2 months ago
- Implementation of Memformer, a Memory-augmented Transformer, in Pytorch☆115Updated 4 years ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆204Updated last year
- Code for the paper, "Distribution Augmentation for Generative Modeling", ICML 2020.☆122Updated last year
- ☆376Updated last year
- [ICML 2021 Oral] We show pure attention suffers rank collapse, and how different mechanisms combat it.☆164Updated 4 years ago
- GPT, but made only out of MLPs☆88Updated 3 years ago
- Code for "Recurrent Independent Mechanisms"☆118Updated 3 years ago
- VQVAEs, GumbelSoftmaxes and friends☆557Updated 3 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆127Updated 2 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 8 months ago
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆133Updated 4 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆80Updated 3 years ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆225Updated 2 years ago
- An implementation of 2021 paper by Geoffrey Hinton: "How to represent part-whole hierarchies in a neural network" in Pytorch.☆58Updated 4 years ago
- ☆77Updated 4 years ago
- Unofficial implementation of Perceiver IO☆120Updated 2 years ago
- MERLOT: Multimodal Neural Script Knowledge Models☆223Updated 3 years ago
- Benchmark for Lifelong learning research☆119Updated 3 years ago
- Code for Multi-Head Attention: Collaborate Instead of Concatenate☆152Updated last year
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆458Updated last year
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆257Updated last year