KindXiaoming / BIMTLinks
Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.
☆172Updated 2 years ago
Alternatives and similar repositories for BIMT
Users that are interested in BIMT are comparing it to the libraries listed below
Sorting:
- The boundary of neural network trainability is fractal☆215Updated last year
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆93Updated 5 months ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆73Updated 2 years ago
- Automatic gradient descent☆210Updated 2 years ago
- ☆228Updated 2 weeks ago
- Omnigrok: Grokking Beyond Algorithmic Data☆61Updated 2 years ago
- Tools for working with the Abstraction & Reasoning Corpus☆201Updated last week
- Learning Universal Predictors☆79Updated last year
- Graph neural networks in JAX.☆67Updated last year
- Repository for code used in the xVal paper☆142Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆87Updated last year
- unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆79Updated 3 years ago
- ☆207Updated 8 months ago
- A State-Space Model with Rational Transfer Function Representation.☆79Updated last year
- ☆52Updated last year
- ☆164Updated last year
- Hierarchical Associative Memory User Experience☆103Updated last month
- ☆56Updated 10 months ago
- Cellular Automata Accelerated in JAX (Oral at ICLR 2025)☆211Updated 3 months ago
- ☆180Updated 2 weeks ago
- The history files when recording human interaction while solving ARC tasks☆114Updated last week
- Getting crystal-like representations with harmonic loss☆194Updated 4 months ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆239Updated last year
- 🧱 Modula software package☆225Updated last week
- Codes for the paper "A mathematical perspective on Transformers".☆38Updated last year
- Generative Flow Networks - GFlowNet☆269Updated last week
- Neural Networks and the Chomsky Hierarchy☆208Updated last year
- Scalable and Stable Parallelization of Nonlinear RNNS☆20Updated this week
- The Energy Transformer block, in JAX☆59Updated last year
- Running Jax in PyTorch Lightning☆113Updated 8 months ago