Differentiable brain modeling.
BrainPy
Modeling of point neuron-based spiking neural networks.
Comes from Prof. Si Wu's lab (Peking University). See BrainPy 1.0 paper, BrainPy 2.0 eLife paper, and our ICLR 2024 paper.
BrainCell
Modeling of biologically detailed Hodgkin-Huxley neurons and networks.
Collaborated with Prof. Songting Li (Shanghai Jiao Tong University).
BrainMass
Whole-brain modeling with differentiable neural mass models.
Shared infrastructure.
BrainUnit
Physical units and unit-aware mathematical system for brain modeling.
See our paper in Nature Communications 2025.
BrainEvent
Event-driven algorithms and data structures for brain dynamics.
BrainTrace
Online learning system for large-scale brain models.
See our paper in Nature Communications 2026.
BrainTools
Modeling tools for general-purpose brain simulation.
JAX-native state runtime.
BrainState
State-based IR compilation for brain models.
Installation.
pip install BrainX[cuda13] -U