NativeSolver

experimental.network_dynamics.solvers.NativeSolver()

Base class for Network Dynamics’s native solvers (manual implementations).

Methods

Name Description
step Single integration step.

step

experimental.network_dynamics.solvers.NativeSolver.step(
    dynamics_fn,
    t,
    state,
    coupling_input,
    dt,
    params,
)

Single integration step.

Args: dynamics_fn: Network dynamics function t: Current time state: Current state, shape [n_states, n_nodes] coupling_input: Coupling input, shape [n_coupling_inputs, n_nodes] dt: Timestep params: Network parameters

Returns: next_state: Updated state, shape [n_states, n_nodes]