pypomp.core.parameters.PanelParameters.to_jax_array

PanelParameters.to_jax_array(param_names: list[str], unit_names: list[str] | None = None, **kwargs) → Array[source]

Converts the parameters to a JAX array suitable for model functions.

Parameters:

  • param_names – A list of canonical parameter names expected by the model.

  • **kwargs – Additional context required for conversion (e.g. unit names).

Returns:

A JAX array representing the parameters. - For Pomp: Shape (reps, n_params) - For PanelPomp: Shape (reps, n_units, n_params)