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.