load_atmodeller_cache#

exo_skryer.vert_chem.load_atmodeller_cache(species_list: list[str], nlay: int, solver_kwargs: dict | None = None) → None[source]#

Build an atmodeller.EquilibriumModel from species_list and cache it globally.

Parameters:

species_listlist of str: Species strings in atmodeller notation, e.g. ['H2O_g', 'CH4_g', 'H2_g', 'He_g']. All gas-phase entries (suffix _g) are registered as retrievable VMR outputs.
nlayint: Number of atmospheric layers. Sets the batch size of the cached Parameters pytree so that eqx.tree_at updates during retrieval match the correct shape.
solver_kwargsdict, optional: Keyword arguments forwarded to SolverParameters. Supported keys: atol, rtol, max_steps, multistart, multistart_perturbation, jac. Defaults (atmodeller’s own) are used for any key not supplied. Reduce multistart and loosen atol/rtol for faster GPU throughput.