borjanG / 2023-transformers-rotfLinks
Codes for the paper "A mathematical perspective on Transformers".
☆39Updated last year
Alternatives and similar repositories for 2023-transformers-rotf
Users that are interested in 2023-transformers-rotf are comparing it to the libraries listed below
Sorting:
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆92Updated last year
- ☆33Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆83Updated last year
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆174Updated 2 years ago
- Code for Discovering Preference Optimization Algorithms with and for Large Language Models☆64Updated last year
- ☆62Updated last year
- About A collection of AWESOME things about information geometry Topics☆175Updated last year
- JAX implementation of Kolmogorov Arnold Networks (KANs).☆10Updated last year
- Codes for the paper The emergence of clusters in self-attention dynamics.☆17Updated 2 years ago
- ☆38Updated last year
- Open source code for EigenGame.☆34Updated 2 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆39Updated 3 years ago
- Omnigrok: Grokking Beyond Algorithmic Data☆62Updated 2 years ago
- unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆81Updated 3 years ago
- The Energy Transformer block, in JAX☆63Updated 2 years ago
- Code for papers Linear Algebra with Transformers (TMLR) and What is my Math Transformer Doing? (AI for Maths Workshop, Neurips 2022)☆76Updated last year
- ☆167Updated 2 years ago
- ☆53Updated last year
- ☆35Updated last year
- Meta-Learning for Compositionality (MLC) for modeling human behavior☆145Updated last month
- ☆41Updated 3 years ago
- ☆35Updated last year
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆63Updated 4 years ago
- A MAD laboratory to improve AI architecture designs 🧪☆136Updated last year
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated 2 years ago
- ☆35Updated last year
- ☆54Updated last year
- Graph neural networks in JAX.☆68Updated last year
- Code for our ACL '23 paper titled "Grokking of Hierarchical Structure in Vanilla Transformers"☆23Updated 2 years ago
- Lightning-like training API for JAX with Flax☆45Updated last year