pypomp.core.parameters.PanelParameters.to_jax_array¶

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

Convert to a JAX array matching the order of param_names and unit_names.

Parameters:

param_names (list[str], optional) – A list of parameter names in the desired order. If None (default), returns the array matching the canonical order of parameters.
unit_names (list[str], optional) – A list of unit names in the desired order. If None (default), returns array for all units.

Returns:

A JAX array of shape (num_theta_idx, n_units, n_params).

Return type:

jax.Array