jcm.model

User-facing Model class.

The Model orchestrates a simulation: forcing, run/resume/run_from_state, chunked op-split scan, post-processing, and xarray conversion. The two component contracts it routes between are:

  • the DynamicalCore (state initialisation, the per-dt step, the gridpoint↔native bridge) — see jcm.dycore;

  • the Physics (parameterizations producing gridpoint PhysicsTendency from a PhysicsState) — see jcm.physics.

The Model itself owns nothing dynamics- or physics-specific; it just threads the cross-step physics carry through the scan, handles the sim-time / date bookkeeping, and produces an xarray trajectory.

Classes

Model([dycore, coords, time_step, terrain, ...])

Top level class for a JAX-GCM simulation.

ModelPredictions(predictions, coords, physics)

User-facing container for model prediction outputs.