CBMOS: a GPU-enabled Python framework for the numerical study of center-based models

Abstract

Cell-based models are becoming increasingly popular for applications in developmental biology. However, the impact of numerical choices on the accuracy and efficiency of the simulation of these models is rarely meticulously tested. We present CBMOS, a Python framework for the simulation of the center-based or cell-centered model. Contrary to other implementations, CBMOS focus is on facilitating numerical study of center-based models by providing access to multiple ODE solvers and force functions through a flexible, user-friendly API. We show-case its potential by evaluating the use of the backward Euler method for calculating the trajectories of two-dimensional cell populations. We confirm that although for moderate accuracy levels the backward Euler method allows for larger time step sizes than the commonly used forward Euler method, its additional computational cost due to being an implicit method prohibits its use for practical test cases. CBMOS is available on GitHub1 and PyPI under an MIT license. It allows for fast prototyping on a CPU for small systems through the use of NumPy. Using CuPy on a GPU, cell populations of up to 10,000 cells can be simulated within a few seconds. As such, we hope it can also be of use to modelers interested in testing preliminary hypotheses before committing to more complex center-based model frameworks. AMS subject classification65Z05, 92C15, 92-10