pypomp.core.parameters.PompParameters.to_jax_array¶

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

Convert to a JAX array matching the order of param_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.
Returns:: A JAX array of shape (num_theta_idx, n_params).
Return type:: jax.Array