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
- Automatic gradient descent☆208Updated 2 years ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆72Updated 2 years ago
- ☆226Updated this week
- Graph neural networks in JAX.☆67Updated last year
- unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆78Updated 3 years ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆93Updated 4 months ago
- Tools for working with the Abstraction & Reasoning Corpus☆197Updated last year
- Learning Universal Predictors☆78Updated last year
- Hierarchical Associative Memory User Experience☆103Updated 3 weeks ago
- Omnigrok: Grokking Beyond Algorithmic Data☆61Updated 2 years ago
- ☆163Updated last year
- Neural Networks and the Chomsky Hierarchy☆207Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆85Updated last year
- The Energy Transformer block, in JAX☆59Updated last year
- ☆63Updated 4 months ago
- ☆174Updated 4 months ago
- 🧱 Modula software package☆216Updated 2 weeks ago
- A State-Space Model with Rational Transfer Function Representation.☆79Updated last year
- ☆51Updated last year
- Running Jax in PyTorch Lightning☆111Updated 7 months ago
- ☆206Updated 8 months ago
- Because we don't have enough time to read everything☆89Updated 10 months ago
- Codes for the paper "A mathematical perspective on Transformers".☆37Updated last year
- Cellular Automata Accelerated in JAX (Oral at ICLR 2025)☆209Updated 2 months ago
- Repository for code used in the xVal paper☆140Updated last year
- Code for 1st place solution to Kaggle's Abstraction and Reasoning Challenge☆159Updated last month
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆290Updated 11 months ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Uncertainty quantification with PyTorch☆367Updated 3 months ago