CE_rate_jax#

exo_skryer.vert_chem.CE_rate_jax(p_lay: Array, T_lay: Array, params: Dict[str, Array], nlay: int) → Dict[str, Array][source]#

Compute chemical equilibrium profiles using the RateJAX solver.

Parameters:
p_layndarray, shape (nlay,)

Layer pressures. In the forward model this is typically in dyne cm⁻² and is converted internally to bar for the solver.

T_layndarray, shape (nlay,)

Layer temperatures in Kelvin.

paramsdict[str, ndarray]

Chemical abundance parameters containing:

  • M_to_Hfloat

    Metallicity relative to solar in dex.

  • C_to_Ofloat

    Carbon-to-oxygen ratio (dimensionless).

nlayint

Number of atmospheric layers (unused; kept for API compatibility).

Returns:
vmr_laydict[str, ndarray]

Dictionary mapping species names to VMR profiles with shape (nlay,).