Skip to content

Differentiable Mars GCM

tform models Mars's spatial atmosphere and surface with a differentiable three-dimensional general-circulation model. The primary workflow uses JAX and Dinosaur to evolve circulation, radiation, surface temperature, regolith and CO₂ frost over MOLA terrain.

Start with installation, then run your first GCM map. For how a simulation is built, read the architecture overview and Mars configuration.

What you can do

Workflow Guide
Generate pressure, temperature, wind and frost maps GCM quickstart
Explore fields and controls in a browser Visualizer
Differentiate and calibrate physical controls Calibration and ablations
Train replacement radiation or bounded heating Neural experiments
Fit a temperature correction to frozen forecasts Temperature-only postprocessing
Compare against Ames, MCD and ARCO-MACDA Reference diagnostics
Assess numerical and scientific evidence Validation and limitations

Implementation

Module Role
src.framework.gcm Body constants, spectral/sigma coordinates, dry dynamics, restart and parameterized integration
src.framework.physics Radiation, surface exchange, regolith, PBL, convection and CO₂ exchange
src.framework.neural Column models, radiation adapters, bounded heating and inference checkpoints
src.celestials.planets.mars Mars constants, forcing factories, terrain/boundary data and map exports
apps/mars-calibration Coupled physical calibration and reproducible experiment protocols
cli and ui Map commands, comparison reports and browser visualization

Software tests and reference diagnostics support specific implementation claims. They do not establish observational accuracy, fully equilibrated climate or predictive high-pressure terraforming behavior. Read the validation guide before interpreting simulated maps as Mars climate predictions.

Deprecated global-mean model

The torch global-mean model and its fast/accurate ODE workflows are deprecated. Use the GCM for new work. Their documentation remains available under Deprecated global-mean model for existing users.