pypomp.PompParameters.to_jax_array¶

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

Convert parameter values to a JAX array matching param_names.

Parameters:

param_names (list of str, optional) – Parameter names in the desired order. If None (default), returns the array in the canonical parameter order.
**kwargs (dict) – Unused in standard pomp parameters.

Returns:

JAX array of shape (n_reps, n_params).

Return type:

jax.Array