Skip to content

NeuroAI-Research/brain-modules

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Intro

My approach to brain simulation

  1. Identify a list of computational modules in the brain
  2. For each module
    1. Understand what it computes: input -> output
    2. Get training data (synthetic or real-world)
    3. Train artificial neural networks to replicate its functionality
  3. Combine modules

List of implemented modules

pip install brain-modules
  1. Place Cells (incomplete, deprecated)
  2. Brain's GPS System (Head Direction Cells, Grid Cells, Place Cells)

List of implemented ANN modules

  1. Transformer (RMSNorm, RoPE, SwiGLU, GroupedQueryAttention)
  2. GRU (Gated Recurrent Unit)
  3. SwiGLUMLP (Swish-Gated Linear Unit MLP)

Releases

No releases published

Packages

 
 
 

Contributors