Brain Simulation Ecosystem

Brain Simulation Ecosystem#

BrainX is an open ecosystem for building, running, and studying brain simulations across scales. It brings together JAX-native libraries for unit-safe computation, sparse event processing, reusable experiment tooling, stateful model construction, biophysical and neural-mass modeling, and learning-driven dynamics. For the project landing page and broader ecosystem overview, see brainx.chaobrain.com.

This page is a summary of the BrainX documentation. It introduces the core packages, shows how the layers fit together, and points you to tutorials that move from foundational tools to complete simulation workflows.

If you are setting up BrainX for the first time, start with the installation tutorial to prepare your environment, install the required packages.

The tutorials are organized around the BrainX package layers:

  • Physical Units : brainunit

  • Events & Tools : brainevent braintools

  • State Runtime : brainstate

  • Models : brainpy.state braincell brainmass

  • Online Learning : braintrace

For more detailed information, click a package card to open its full package documentation:

brainunit

Unit-aware numerical computing for values, parameters, and equations.

https://brainx.chaobrain.com/brainunit/
brainevent

Event-driven computation for sparse spike-based workloads.

https://brainx.chaobrain.com/brainevent/
braintools

Experiment utilities for inputs, tasks, connectivity, metrics, and plots.

https://brainx.chaobrain.com/braintools/
brainstate

Stateful JAX modules, parameters, transforms, and runtime management.

https://brainx.chaobrain.com/brainstate/
brainpy.state

Point-neuron networks, synapses, projections, and SNN models.

https://brainx.chaobrain.com/brainpy-state/
braincell

Biophysical cell modeling with ion channels and morphology support.

https://brainx.chaobrain.com/braincell/
brainmass

Neural mass models and whole-brain dynamics.

https://brainx.chaobrain.com/brainmass/
braintrace

Online learning and eligibility traces for RNNs and SNNs.

https://brainx.chaobrain.com/braintrace/