prepare

experimental.network_dynamics.prepare(
    dynamics,
    solver,
    t0=0.0,
    t1=1.0,
    dt=0.1,
    n_nodes=1,
    noise=None,
    externals=None,
)

Compile a model into a pure JAX solve function and a config PyTree.

Builds per-dispatch data (coupling buffers, noise samples, external inputs) and returns (solve_fn, config) where solve_fn(config) runs the integration. Dispatches on the first two arguments via plum: Network/AbstractDynamics paired with NativeSolver/DiffraxSolver.

Parameters

Name	Type	Description	Default
t0	float	Integration interval and step size. dt is the fixed step for native solvers and the initial step for Diffrax.	0.0
t1	float	Integration interval and step size. dt is the fixed step for native solvers and the initial step for Diffrax.	0.0
dt	float	Integration interval and step size. dt is the fixed step for native solvers and the initial step for Diffrax.	0.0

Returns

Name	Type	Description
	(Callable, Bunch)	Pure solve function and its runtime configuration PyTree.
	See help(prepare) or prepare.__doc__ for the full reference,
	including per-dispatch parameters (n_nodes, noise, externals
	for bare dynamics) and Diffrax limitations (no delays, no auxiliaries,
	no VOI filtering).