KindXiaoming / BIMTLinks
Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.
☆171Updated 2 years ago
Alternatives and similar repositories for BIMT
Users that are interested in BIMT are comparing it to the libraries listed below
Sorting:
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆72Updated 2 years ago
- Omnigrok: Grokking Beyond Algorithmic Data☆58Updated 2 years ago
- σ-GPT: A New Approach to Autoregressive Models☆65Updated 10 months ago
- ☆51Updated last year
- Oscillatory State-Space Models☆88Updated 3 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆87Updated 3 months ago
- unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆78Updated 2 years ago
- The Energy Transformer block, in JAX☆58Updated last year
- Graph neural networks in JAX.☆67Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆78Updated last year
- Neural Networks and the Chomsky Hierarchy☆205Updated last year
- Hierarchical Associative Memory User Experience☆100Updated last year
- Pytorch-like dataloaders for JAX.☆87Updated last month
- Scalable and Stable Parallelization of Nonlinear RNNS☆16Updated 5 months ago
- Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊☆123Updated 3 weeks ago
- The boundary of neural network trainability is fractal☆208Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆83Updated last year
- Flow-matching algorithms in JAX☆97Updated 10 months ago
- ☆174Updated last year
- ☆53Updated 8 months ago
- ☆54Updated last year
- ☆26Updated 2 years ago
- Automatic gradient descent☆208Updated 2 years ago
- Generative Flow Networks - GFlowNet☆255Updated this week
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆91Updated 2 weeks ago
- Codes for the paper "A mathematical perspective on Transformers".☆37Updated 11 months ago
- Cellular Automata Accelerated in JAX (Oral at ICLR 2025)☆204Updated last month
- ☆81Updated last year
- ☆39Updated 3 years ago
- 🧱 Modula software package☆200Updated 3 months ago