Runge-Kutta ODE Integrator Implemented in Cython and Numba
CyRK provides fast integration tools to solve systems of ODEs using an adaptive time stepping scheme. CyRK can accept differential equation functions that are written in pure Python or njited numba, speeding up development time. The purpose of this package is to provide some functionality of scipy's solve_ivp with improved performance.
Currently, CyRK's numba (njit-safe) implementation is 10-100x faster than scipy's solve_ivp function. The cython implementation is 5-20x faster. The cython function is also largely pre-compiled which avoids most of the initial performance hit found with using the numba version.
It is recommended you use an Anaconda environment. CyRK has been tested on Python 3.8--3.10
To install simply open a terminal and call:
pip install CyRK
If not installing from a wheel, CyRK will attempt to install Cython
and Numpy
in order to compile the cython code.
After the files have been compiled, cython will be uninstalled and CyRK's runtime dependencies
(see the pyproject.toml file for the latest list) will be installed instead.
A new installation of CyRK can be tested quickly by running the following from a python console.
from CyRK import test_cyrk, test_nbrk
test_cyrk()
# You will hopefully see the message "CyRK's cyrk_ode was tested successfully."
test_nbrk()
# You will hopefully see the message "CyRK's nbrk_ode was tested successfully."
Please report installation issues. We will work on a fix and/or add workaround information here.
If you see a "Can not load module: CyRK.cy" or similar error then the cython extensions likely did not compile during installation. Try running pip install CyRK --no-binary="CyRK"
to force python to recompile the cython extensions locally (rather than via a prebuilt wheel).
If you intend to work on CyRK's code base you will want to install the following dependencies in order to run CyRK's test suite.
conda install pytest scipy matplotlib jupyter
conda install
can be replaced with pip install
if you prefer.
CyRK's API is similar to SciPy's solve_ivp function. A differential equation can be defined in python such as:
import numpy as np
from numba import njit
# For even more speed up you can use numba's njit to compile the diffeq
@njit
def diffeq_nb(t, y):
dy = np.empty_like(y)
dy[0] = (1. - 0.01 * y[1]) * y[0]
dy[1] = (0.02 * y[0] - 1.) * y[1]
return dy
initial_conds = np.asarray((20., 20.), dtype=np.complex128)
time_span = (0., 50.)
rtol = 1.0e-7
atol = 1.0e-8
The ODE can then be solved using the numba function by calling CyRK's nbrk_ode
:
from CyRK import nbrk_ode
time_domain, y_results, success, message = \
nbrk_ode(diffeq_nb, time_span, initial_conds, rk_method=1, rtol=rtol, atol=atol)
To call the cython version of the integrator you need to slightly edit the differential equation so that it does not return the derivative. Instead, the output is passed as an input argument (a np.ndarray) to the function.
@njit
def diffeq_cy(t, y, dy):
dy[0] = (1. - 0.01 * y[1]) * y[0]
dy[1] = (0.02 * y[0] - 1.) * y[1]
Alternatively, you can use CyRK's conversion helper functions to automatically convert between numba/scipy and cyrk function calls.
from CyRK import nb2cy, cy2nb
@njit
def diffeq_nb(t, y):
dy = np.empty_like(y)
dy[0] = (1. - 0.01 * y[1]) * y[0]
dy[1] = (0.02 * y[0] - 1.) * y[1]
return dy
diffeq_cy = nb2cy(diffeq_nb, use_njit=True)
diffeq_nb2 = cy2nb(diffeq_cy, use_njit=True)
You can then call the ODE solver in a similar fashion as the numba version.
from CyRK import cyrk_ode
time_domain, y_results, success, message = \
cyrk_ode(diffeq_cy, time_span, initial_conds, rk_method=1, rtol=rtol, atol=atol)
Both the numba and cython versions of the ODE solver have the following optional inputs:
rtol
: Relative Tolerance (default is 1.0e-6).atol
: Absolute Tolerance (default is 1.0e-8).max_step
: Maximum step size (default is +infinity).first_step
: Initial step size (default is 0).- If 0, then the solver will try to determine an ideal value.
args
: Python tuple of additional arguments passed to thediffeq
.t_eval
: Both solvers uses an adaptive time stepping protocol based on the recent error at each step. This results in a final non-uniform time domain of variable size. If the user would like the results at specific time steps then they can provide a np.ndarray array at the desired steps tot_eval
. The solver will then interpolate the results to fit this array.rk_method
: Runge-Kutta method (default is 1; all of these methods are based off of SciPy implementations):0
- "RK23" Explicit Runge-Kutta method of order 3(2).1
- "RK45" Explicit Runge-Kutta method of order 5(4).2
- "DOP853" Explicit Runge-Kutta method of order 8.
capture_extra
andinterpolate_extra
: CyRK has the capability of capturing additional parameters during integration. Please seeDocumentation\Extra Output.md
for more details.
- Issue 1: Absolute tolerance can only be passed as a single value (same for all y's).
- Issue 3: Right now the cython version only allows for complex-valued y-values.
It is great to see CyRK used in other software or in scientific studies. We ask that you cite back to CyRK's GitHub website so interested parties can learn about this package.
Renaud, Joe P. (2022). CyRK - ODE Integrator Implemented in Cython and Numba. Zenodo. https://doi.org/10.5281/zenodo.7093266
In addition to citing CyRK, please consider citing SciPy and its references for the specific Runge-Kutta model that was used in your work. CyRK is largely an adaptation of SciPy's functionality. Find more details here.
Pauli Virtanen, Ralf Gommers, Travis E. Oliphant, Matt Haberland, Tyler Reddy, David Cournapeau, Evgeni Burovski, Pearu Peterson, Warren Weckesser, Jonathan Bright, Stéfan J. van der Walt, Matthew Brett, Joshua Wilson, K. Jarrod Millman, Nikolay Mayorov, Andrew R. J. Nelson, Eric Jones, Robert Kern, Eric Larson, CJ Carey, İlhan Polat, Yu Feng, Eric W. Moore, Jake VanderPlas, Denis Laxalde, Josef Perktold, Robert Cimrman, Ian Henriksen, E.A. Quintero, Charles R Harris, Anne M. Archibald, Antônio H. Ribeiro, Fabian Pedregosa, Paul van Mulbregt, and SciPy 1.0 Contributors. (2020) SciPy 1.0: Fundamental Algorithms for Scientific Computing in Python. Nature Methods, 17(3), 261-272.
Please look here for an up-to-date list of contributors to the CyRK package.
CyRK is open-source and is distributed under the Creative Commons Attribution-ShareAlike 4.0 International license.
You are welcome to fork this repository and make any edits with attribution back to this project (please see the
Citing CyRK
section).
- We encourage users to report bugs or feature requests using GitHub Issues.
- If you would like to contribute but don't know where to start, check out the good first issue tag on GitHub.
- Users are welcome to submit pull requests and should feel free to create them before the final code is completed so that feedback and suggestions can be given early on.