KindXiaoming / BIMT
Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.
☆164Updated last year
Alternatives and similar repositories for BIMT:
Users that are interested in BIMT are comparing it to the libraries listed below
- Omnigrok: Grokking Beyond Algorithmic Data☆52Updated last year
- Automatic gradient descent☆207Updated last year
- Graph neural networks in JAX.☆67Updated 8 months ago
- unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆72Updated 2 years ago
- Reverse Engineering the Abstraction and Reasoning Corpus☆229Updated 4 months ago
- Learning Universal Predictors☆74Updated 6 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆53Updated 2 months ago
- ☆201Updated this week
- The boundary of neural network trainability is fractal☆194Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆102Updated 2 months ago
- Generative Flow Networks - GFlowNet☆226Updated 3 months ago
- ☆36Updated 2 months ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated last year
- ☆158Updated 2 months ago
- ☆49Updated last year
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)☆185Updated 8 months ago
- The Energy Transformer block, in JAX☆56Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆77Updated 9 months ago
- LoRA for arbitrary JAX models and functions☆135Updated 11 months ago
- Hierarchical Associative Memory User Experience☆98Updated last year
- Implementation of PSGD optimizer in JAX☆28Updated last month
- Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊☆120Updated 3 weeks ago
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆144Updated 2 months ago
- SR based on LLMs.☆94Updated 2 years ago
- ☆149Updated 6 months ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆69Updated last year
- 🧱 Modula software package☆145Updated this week
- Domain Specific Language for the Abstraction and Reasoning Corpus☆236Updated 4 months ago
- Neural Networks and the Chomsky Hierarchy☆199Updated 10 months ago
- ☆26Updated last year