BrainX ecosystem packages
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.
Collaboration with Prof. Douglas Zhou and Prof. 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.
BrainTrace
Online learning system for large-scale brain models.
See our paper in Nature Communications 2026.
BrainEvent
Event-driven algorithms and data structures for brain dynamics.
BrainTools
Modeling tools for general-purpose brain simulation.
JAX-native state runtime.
BrainState
State-based IR compilation for brain models.
Vibe coding.
BrainX skill
Agent skills that teach your coding assistant BrainX.
Installation.
BrainX
CUDA 12
CUDA 13
BrainX skill
cd /actual/path/to/project npx skills add chaobrain/BrainX-skill --agent codex