English

TorchCor: High-Performance Cardiac Electrophysiology Simulations with the Finite Element Method on GPUs

Software Engineering 2025-10-15 v1

Abstract

Cardiac electrophysiology (CEP) simulations are increasingly used for understanding cardiac arrhythmias and guiding clinical decisions. However, these simulations typically require high-performance computing resources with numerous CPU cores, which are often inaccessible to many research groups and clinicians. To address this, we present TorchCor, a high-performance Python library for CEP simulations using the finite element method on general-purpose GPUs. Built on PyTorch, TorchCor significantly accelerates CEP simulations, particularly for large 3D meshes. The accuracy of the solver is verified against manufactured analytical solutions and the NN-version benchmark problem. TorchCor is freely available for both academic and commercial use without restrictions.

Keywords

Cite

@article{arxiv.2510.12011,
  title  = {TorchCor: High-Performance Cardiac Electrophysiology Simulations with the Finite Element Method on GPUs},
  author = {Bei Zhou and Maximilian Balmus and Cesare Corrado and Ludovica Cicci and Shuang Qian and Steven A. Niederer},
  journal= {arXiv preprint arXiv:2510.12011},
  year   = {2025}
}