LumenPallidium / energy_transformerLinks
Pytorch implementation of an energy transformer - an energy-based reccurrent variant of the transformer.
☆13Updated last year
Alternatives and similar repositories for energy_transformer
Users that are interested in energy_transformer are comparing it to the libraries listed below
Sorting:
- The Energy Transformer block, in JAX☆58Updated last year
- ☆39Updated 3 years ago
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆24Updated last year
- Code associated to papers on superposition (in ML interpretability)☆28Updated 2 years ago
- Evaluation of neuro-symbolic engines☆35Updated 10 months ago
- This repository has implementations of various alternatives to backpropagation for training neural networks.☆22Updated 5 months ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Flexible Inference for Predictive Coding Networks in JAX.☆48Updated 3 weeks ago
- Learning Universal Predictors☆76Updated 10 months ago
- Brain-like variational inference☆51Updated last month
- Understanding how features learned by neural networks evolve throughout training☆35Updated 8 months ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Code repository for "The Clock and the Pizza: Two Stories in Mechanistic Explanation of Neural Networks"☆17Updated last year
- Example of Dense Associative Memory training on MNIST☆36Updated 2 years ago
- Omnigrok: Grokking Beyond Algorithmic Data☆58Updated 2 years ago
- We study toy models of skill learning.☆28Updated 5 months ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆32Updated 3 years ago
- ☆26Updated 2 years ago
- ☆53Updated 8 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆87Updated 3 months ago
- ☆17Updated 2 years ago
- Universal Neurons in GPT2 Language Models☆29Updated last year
- Official repository for the paper "Exploring the Promise and Limits of Real-Time Recurrent Learning" (ICLR 2024)☆10Updated 2 weeks ago
- Official code for the ICML 2024 paper "The Entropy Enigma: Success and Failure of Entropy Minimization"☆52Updated last year
- unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆78Updated 2 years ago
- Tools for exploring Transformer neuron behaviour, including input pruning and diversification.☆20Updated last year
- ☆32Updated 8 months ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆74Updated 7 months ago
- Official implementation of the transformer (TF) architecture suggested in a paper entitled "Looped Transformers as Programmable Computers…☆27Updated 2 years ago