glassroom / heinsen_routingLinks
Reference implementation of "An Algorithm for Routing Vectors in Sequences" (Heinsen, 2022) and "An Algorithm for Routing Capsules in All Domains" (Heinsen, 2019), for composing deep neural networks.
☆169Updated 2 years ago
Alternatives and similar repositories for heinsen_routing
Users that are interested in heinsen_routing are comparing it to the libraries listed below
Sorting:
- Official repository for the paper "A Modern Self-Referential Weight Matrix That Learns to Modify Itself" (ICML 2022 & NeurIPS 2021 Deep R…☆173Updated last month
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆94Updated 7 months ago
- ☆252Updated 2 years ago
- Fast Text Classification with Compressors dictionary☆150Updated last year
- An implementation of the Nyströmformer, using Nystrom method to approximate standard self attention☆56Updated 2 years ago
- Official Repository of Pretraining Without Attention (BiGS), BiGS is the first model to achieve BERT-level transfer learning on the GLUE …☆113Updated last year
- ☆143Updated 2 years ago
- Experiments for efforts to train a new and improved t5☆76Updated last year
- Amos optimizer with JEstimator lib.☆82Updated last year
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆59Updated 3 years ago
- A playground to make it easy to try crazy things☆33Updated last month
- A case study of efficient training of large language models using commodity hardware.☆68Updated 2 years ago
- Controlled Text Generation via Language Model Arithmetic☆222Updated 10 months ago
- [NAACL 2025] Official Implementation of "HMT: Hierarchical Memory Transformer for Long Context Language Processing"☆72Updated 3 weeks ago
- ☆68Updated 11 months ago
- A tool to analyze and debug neural networks in pytorch. Use a GUI to traverse the computation graph and view the data from many different…☆287Updated 7 months ago
- Implementation of Memorizing Transformers (ICLR 2022), attention net augmented with indexing and retrieval of memories using approximate …☆634Updated 2 years ago
- This repo contains code for the paper: "Can Foundation Models Help Us Achieve Perfect Secrecy?"☆24Updated 2 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- ☆20Updated 3 years ago
- A library for incremental loading of large PyTorch checkpoints☆56Updated 2 years ago
- Automatic gradient descent☆208Updated 2 years ago
- TART: A plug-and-play Transformer module for task-agnostic reasoning☆200Updated 2 years ago
- My explorations into editing the knowledge and memories of an attention network☆35Updated 2 years ago
- RWKV model implementation☆38Updated 2 years ago
- The GeoV model is a large langauge model designed by Georges Harik and uses Rotary Positional Embeddings with Relative distances (RoPER).…☆121Updated 2 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆239Updated 2 years ago
- An interactive exploration of Transformer programming.☆265Updated last year
- Implements the Tsetlin Machine, Coalesced Tsetlin Machine, Convolutional Tsetlin Machine, Regression Tsetlin Machine, and Weighted Tsetli…☆148Updated 2 months ago
- An interpreter for RASP as described in the ICML 2021 paper "Thinking Like Transformers"☆316Updated 10 months ago