rmwkwok / forward_forward_algorithmLinks
Implementation of Geoffrey Hinton's forward forward algorithm in tensorflow
☆33Updated 2 years ago
Alternatives and similar repositories for forward_forward_algorithm
Users that are interested in forward_forward_algorithm are comparing it to the libraries listed below
Sorting:
- Implementation/simulation of the predictive forward-forward credit assignment algorithm for training neurobiologically-plausible recurren…☆58Updated 2 years ago
- Implementation of the transformer proposed in "Building Blocks for a Complex-Valued Transformer Architecture"☆77Updated last year
- This notebook is designed to plot the attention maps of a vision transformer trained on MNIST digits.☆36Updated last week
- ☆74Updated 2 years ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆100Updated 5 months ago
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆119Updated 7 months ago
- ☆51Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆78Updated last year
- FID computation in Jax/Flax.☆27Updated 10 months ago
- PyTorch implementation of Hinton's FF Algorithm with hard negatives sampling☆14Updated 2 years ago
- Trying out the Mamba architecture on small examples (cifar-10, shakespeare char level etc.)☆46Updated last year
- Spyx: Spiking Neural Networks in JAX☆122Updated 7 months ago
- Reimplementation of Geoffrey Hinton's Forward-Forward Algorithm☆148Updated last year
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Updated 2 years ago
- Implementation of Agent Attention in Pytorch☆90Updated 10 months ago
- ☆31Updated last year
- A repository for log-time feedforward networks☆222Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆82Updated last year
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆106Updated 6 months ago
- The boundary of neural network trainability is fractal☆204Updated last year
- Implementation of Block Recurrent Transformer - Pytorch☆218Updated 9 months ago
- Evaluating the Mamba architecture on the Othello game☆47Updated last year
- Induce brain-like topographic structure in your neural networks☆62Updated 2 weeks ago
- Pytorch (Lightning) implementation of the Mamba model☆28Updated last month
- Training small GPT-2 style models using Kolmogorov-Arnold networks.☆117Updated last year
- A repo based on XiLin Li's PSGD repo that extends some of the experiments.☆14Updated 8 months ago
- ☆147Updated last year
- Open source code for EigenGame.☆30Updated 2 years ago
- Implementation of "Gradients without backpropagation" paper (https://arxiv.org/abs/2202.08587) using functorch☆109Updated last year
- Omnigrok: Grokking Beyond Algorithmic Data☆58Updated 2 years ago