脑仿真基础软件

BrainX——实验室自主研发的开源脑仿真软件生态,覆盖从离子通道到全脑网络的多尺度建模,原生支持异构计算硬件,目标支撑数千亿神经元级别的全脑动力学仿真。

BrainX:一套面向全脑动力学仿真的软件生态

BrainX 是实验室自主研发的开源脑仿真软件生态,覆盖从离子通道、单细胞、神经环路到全脑网络的多尺度建模能力。其设计目标是把现代 AI 编译技术(JAX/XLA)引入计算神经科学,让脑动力学仿真原生支持 CPU/GPU/TPU 等异构加速硬件,并与深度学习生态无缝互通,最终把全脑动力学仿真规模推向数千亿神经元级别。

BrainX:一套面向全脑动力学仿真的软件生态

BrainX 模块矩阵

BrainX 按职责分层组合:底部是运行时与数值基础,中部是事件计算与多尺度建模,上层是学习算法与工具集。每一层皆可独立使用、共同协作。

技术架构与关键特性

JAX 原生运行时
基于 BrainState 的状态变换与 JIT 编译,原生支持 CPU、GPU、TPU 与异构计算系统,自动并行调度。
端到端可微仿真
整套仿真过程可微,模型参数可通过梯度方法直接拟合实验数据,连接脑仿真与类脑计算。
单位感知科学计算
BrainUnit / SAIUnit 将物理单位嵌入高性能 AI 计算流水线,从根本上避免量纲错误,提升模型严谨性。
大尺度并行架构
通过 BrainEvent 事件驱动稀疏算子、BrainMass 神经质量降阶以及分布式调度,向 10¹¹ 神经元规模推进。

支持的大尺度脑仿真任务

  • 单细胞精细建模(多腔室、HH、离子通道)
  • 皮层柱与微环路仿真
  • 全脑区脉冲网络仿真
  • 神经质量场与平均场建模
  • 可微参数估计与模型校准
  • 在线学习与任务驱动训练
  • 多尺度耦合仿真

BrainX 既支撑下一板块的建模算法研究,也为第三板块的全脑计算模型提供运行时基底。它是实验室「软件—算法—模型」三位一体研究范式中的底座层。