APIΒΆ

jcm.model

User-facing Model class.

jcm.model.Model([dycore, coords, time_step, ...])

Top level class for a JAX-GCM simulation.

jcm.physics_interface

Dycore-agnostic physics interface types and helpers.

jcm.physics_interface.PhysicsState(u_wind, ...)

Represents the state of the atmosphere in physical (nodal) space.

jcm.physics_interface.PhysicsTendency(...[, ...])

Represents the tendencies (rates of change) of physical variables.

jcm.physics_interface.Physics()

jcm.checkpoint

Model state checkpointing for long, preemptible runs.