Brain Simulation Ecosystem (BrainX)
Overview
The BrainX ecosystem provides a comprehensive framework for brain simulation and modeling. It provides tools and libraries for researchers to model, simulate, train, and analyze neural systems at different scales.
Core components in this ecosystem includes:
-
BrainPy: Modeling of point neuron-based spiking neural networks (SNNs), comes from Prof. Si Wu's lab at Peking University.
-
BrainUnit: Comprehensive physical units and unit-aware mathematical system for brain dynamics.
-
BrainCell: Intuitive, parallel, and efficient simulation for biologically detailed brain cell modeling. Collaborated with Prof. Songting Li's lab at Shanghai Jiao Tong University.
-
BrainMass: Whole-brain modeling with differentiable neural mass models.
-
BrainState: State-based IR compilation for efficient simulation of brain models on CPUs, GPUs, and TPUs.
-
BrainTaichi: The first-generation framework for customizing event-driven operators based on Taichi Lang syntax.
-
BrainEvent: Enabling event-driven computations in brain dynamics.
-
BrainTrace: Eligibility trace-based online learning for brain dynamics: $O(N)$ complexity for SNNs and $O(N^2)$ for RNN computations.
-
BrainTools: Commonly used tools for brain dynamics programming, for example checkpointing.
-
PINNx: Physics-informed neural networks for scientific machine learning in JAX.
-
More components may be added in the future.
Installation
The ecosystem can be installed with the following command:
pip install BrainX -U
This command installs the core package and pins specific versions of the component projects known to work together, ensuring compatibility based on integration tests.
On CPU platforms, the following command can be used to install the ecosystem with all components:
pip install BrainX[cpu] -U
On GPU platforms, the following command can be used to install the ecosystem with all components:
pip install BrainX[cuda12] -U
pip install BrainX[cuda13] -U
On TPU platforms, the following command can be used to install the ecosystem with all components:
pip install BrainX[tpu] -U
For development, you might want to clone the repository and install it in editable mode:
git clone https://github.com/chaobrain/brainx.git
cd brainx
pip install -e .
Documentation
For detailed documentation, tutorials, and examples, visit our Documentation Portal.
Contributing
We welcome contributions from the community! Please see our Contributing Guidelines for more information on how to get involved.
License
This project is licensed under the Apache License, Version 2.0. See the LICENSE file for details.
Citation
If you use the BrainX Ecosystem in your research, please cite it appropriately. Refer to the citation guide on our documentation portal.
Support
If you have questions, encounter issues, or need support, please:
- Check the documentation.
- Search the existing issues.
- Open a new issue if your problem is not addressed.
- Contact us via email:
chao.brain@qq.com.
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file brainx-2026.7.9.tar.gz.
File metadata
- Download URL: brainx-2026.7.9.tar.gz
- Upload date:
- Size: 13.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.1.0 CPython/3.13.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
9360f76d175bdd26d80ac7d699c3df443b2df424b1f17dfa073eb6011b212929
|
|
| MD5 |
e08a05320a379700865d257310ee0486
|
|
| BLAKE2b-256 |
bf5a79abdaa9e00d7a67f0bed4302fb6f7bd59be8020644c6921534add7881e1
|
File details
Details for the file brainx-2026.7.9-py3-none-any.whl.
File metadata
- Download URL: brainx-2026.7.9-py3-none-any.whl
- Upload date:
- Size: 13.9 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.1.0 CPython/3.13.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0d6b1f12424a2b22f349fe1dd34a019299de2cdb2795a751004d9ac02995c30a
|
|
| MD5 |
bd601a2cef707690eee7f22045e548db
|
|
| BLAKE2b-256 |
af190616d1d8617d96122dd647d61de238d3377a9c5f0ee0b1e3c8757b608610
|
