HelioX - A GPU-Native Framework for Simulation and Training of Biophysically Detailed Networks

Accepted at ICML 2026.

HelioX is a GPU-native framework for simulation and training of biophysically detailed neural networks. It targets the mismatch between irregular dendritic computation and conventional deep learning stacks, and provides an end-to-end pipeline for both forward simulation and gradient-based training.

Highlights:

• Multi-stream GPU execution for ionic current calculation, ODE construction, and conductance update
• Efficient spike handling path designed for sparse spike events
• End-to-end differentiable training for deep biophysical neural architectures
• Strong runtime and scalability gains over traditional simulators in large-scale settings