https://github.com/chaobrain/brainevent
Event-driven Computation for Brain Dynamics.
https://github.com/chaobrain/brainevent
brain-dynamics-modeling brain-inspired-computing dynamic-sparsity event-driven-computation sparse-matrix
Last synced: 3 months ago
JSON representation
Event-driven Computation for Brain Dynamics.
- Host: GitHub
- URL: https://github.com/chaobrain/brainevent
- Owner: chaobrain
- License: apache-2.0
- Created: 2025-01-24T09:21:55.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2026-03-25T08:09:20.000Z (3 months ago)
- Last Synced: 2026-03-26T09:29:43.901Z (3 months ago)
- Topics: brain-dynamics-modeling, brain-inspired-computing, dynamic-sparsity, event-driven-computation, sparse-matrix
- Language: Python
- Homepage: https://brainevent.readthedocs.io/
- Size: 4.28 MB
- Stars: 8
- Watchers: 3
- Forks: 3
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- Changelog: changelog.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
- Code of conduct: CODE_OF_CONDUCT.md
- Security: SECURITY.md
Awesome Lists containing this project
README
# Enabling Event-driven Computation in CPU/GPU/TPU
Brain is characterized by the discrete spiking events, which are the fundamental units of computation in the brain.
`BrainEvent` provides a set of data structures and algorithms for such event-driven computation on
**CPUs**, **GPUs**, **TPUs**, and maybe more, which can be used to model the brain dynamics in an
efficient and biologically plausible way.
Particularly, it provides the following class to represent binary events in the brain:
- ``BinaryArray``: representing array with a vector/matrix of events.
Furthermore, it implements the following commonly used data structures for event-driven computation
of the above class:
- ``COO``: a sparse matrix in COO format for sparse and event-driven computation.
- ``CSR``: a sparse matrix in CSR format for sparse and event-driven computation.
- ``CSC``: a sparse matrix in CSC format for sparse and event-driven computation.
- ``JITCHomoR``: a just-in-time connectivity matrix with homogenous weight for sparse and event-driven computation.
- ``JITCNormalR``: a just-in-time connectivity matrix with normal distribution weight for sparse and event-driven
computation.
- ``JITCUniformR``: a just-in-time connectivity matrix with uniform distribution weight for sparse and event-driven
computation.
- ``FixedPreNumConn``: a fixed number of pre-synaptic connections for sparse and event-driven computation.
- ``FixedPostNumConn``: a fixed number of post-synaptic connections for sparse and event-driven computation.
- ...
`BrainEvent` is fully compatible with physical units and unit-aware computations provided
in [BrainUnit](https://github.com/chaobrain/brainunit).
## Usage
If you want to take advantage of event-driven computations, you must warp your data with ``brainevent.BinaryArray``:
```python
import brainevent
# wrap your array with BinaryArray
event_array = brainevent.BinaryArray(your_array)
```
Then, the matrix multiplication with the following data structures, $\mathrm{event\ array} @ \mathrm{data}$,
will take advantage of event-driven computations:
- Sparse data structures provided by ``brainevent``, like:
- ``brainevent.CSR``
- ``brainevent.JITCHomoR``
- ``brainevent.FixedPostNumConn``
- ...
- Dense data structures provided by JAX/NumPy, like:
- ``jax.numpy.ndarray``
- ``numpy.ndarray``
```python
data = jax.random.rand(...) # normal dense array
data = brainevent.CSR(...) # CSR structure
data = brainevent.JITCScalarR(...) # JIT connectivity
data = brainevent.FixedPostNumConn(...) # fixed number of post-synaptic connections
# event-driven matrix multiplication
r = event_array @ data
r = data @ event_array
```
## Installation
You can install ``brainevent`` via pip:
```bash
pip install brainevent -U
```
Alternatively, you can install `BrainX`, which bundles `brainevent` with other compatible packages for a comprehensive brain modeling ecosystem:
```bash
pip install BrainX -U
```
## Documentation
The official documentation is hosted on Read the Docs: [https://brainevent.readthedocs.io/](https://brainevent.readthedocs.io/)
## See also the ecosystem
``brainevent`` is one part of our brain modeling ecosystem: https://brainmodeling.readthedocs.io/