Differentiable brain modeling.
BrainPy
Brain dynamics programming in Python.
Comes from Prof. Si Wu's lab (Peking University). See BrainPy 1.0 paper, BrainPy 2.0 eLife paper, and our ICLR 2024 paper.
brainpy.state
Modeling of point neuron-based spiking neural networks, rebuilt on the BrainX with the state-based syntax.
BrainCell
Modeling of biologically detailed Hodgkin-Huxley neurons and networks.
Collaboration with Prof. Douglas Zhou and Songting Li.
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