earthsci-astdiff (Python)

Python binding of EarthSciASTDiff: analytical sparse Jacobians for .esm models, by rule-based differentiation of the expression AST. Implements the derivative semantics of esm-jacobian-spec.md (§3 per-operator rules, §4 band entries incl. contracted dimensions) against the earthsci-ast types, and is gated by the same band-list goldens as the Julia reference (tests/goldens/*.bands.json, byte-identical deterministic JSON).

import earthsci_ast
from earthsci_astdiff import jacobian_bands, serialize_band, stable_json

file = earthsci_ast.load_path("model.esm")
entries = jacobian_bands(file, "MyModel")  # ∂f/∂u as Band entries
entries_p = jacobian_bands(file, "MyModel", wrt="parameters")
text = stable_json([serialize_band(e) for e in entries])

Scope: the band calculus (jacobian_bands), the §4 entry serialization (serialize_band / parse_band / stable_json), the structural sparsity pattern (jacobian_pattern, plain COO index lists — no scipy dependency), jacobian_block / parse_jacobian_block with the §4.1 template CSE (cse_templates / expand_templates), the structure classification, the factorization plans (per-cell block LU, Markowitz ordering + fill-in), the derived evaluation document (evaluation_document, with hoist_observed), the product documents (jvp_document / vjp_document / hvp_document, spec §9) and the scatter plans (scatter_plan / bind_plan, spec §8) — each byte-identical to the reference's goldens.

The package evaluates nothing: every value is EarthSciAST's, from running one of these documents. The tests do exactly that (tests/docvals.py, with EarthSciAST's numpy interpreter) and gate the results against the Julia reference's value goldens (tests/goldens/*.jvals.json, *.hvals.json; exact key set + 1e-10 relative).