constant_vmr

exo_skryer.vert_chem.constant_vmr(species_order: tuple[str, ...])

Build a JIT-optimized function for constant VMR profiles.

This function creates a chemistry kernel that generates constant (vertically uniform) volume mixing ratio profiles from logarithmic abundance parameters. The returned kernel is optimized for JAX JIT compilation by using a fixed species list determined at build time.

Parameters:

species_ordertuple of str

Ordered tuple of trace species names (e.g., (‘H2O’, ‘CH4’, ‘CO’)). For each species, the kernel will expect a parameter named ‘log_10_f_<species>’ in the params dictionary.

Returns:

callable

A chemistry kernel function with signature: kernel(p_lay, T_lay, params, nlay) -> Dict[str, jnp.ndarray]

The kernel takes: - p_lay : Layer pressures (unused but kept for API compatibility) - T_lay : Layer temperatures (unused but kept for API compatibility) - params : Dictionary containing ‘log_10_f_<species>’ values - nlay : Number of atmospheric layers

And returns a dictionary mapping species names to their VMR profiles.