astronomix.analysis_helpers._exposed_rhs

astronomix.analysis_helpers._exposed_rhs#

Exposed right-hand side of the fluid equations for analysis.

Recomputes the semi-discrete RHS dU/dt = R(U) of the finite-difference hydrodynamics solver as a standalone, differentiable function of the conserved state. This is what the Jacobian helpers linearise; it is deliberately kept outside the time-stepping machinery so it can be fed to jax.jvp directly.