jcm.model

Functions

averaged_trajectory_from_step(step_fn, ...)

Return a function that accumulates repeated applications of step_fn.

Classes

DiagnosticsCollector(*args, **kwargs)
Model(coords[, time_step, terrain, physics, ...])	Top level class for a JAX-GCM configuration using the Speedy physics on an aquaplanet.
Predictions(dynamics, physics, times)