API

jcm.model	User-facing Model class.
jcm.model.Model([dycore, coords, time_step, ...])	Top level class for a JAX-GCM simulation.
jcm.dycore	Pluggable dynamical-core backends for jax-gcm.
jcm.dycore.base.DynamicalCore()	Protocol for a swappable dynamical-core backend.
jcm.dycore.dinosaur.DinosaurDycore(coords, ...)	Spectral dynamical core backed by the dinosaur package.
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.