RS-GW-JAX: Differentiable Randall-Sundrum-Goldberger-Wise Solver
Abstract
This is the initial stable release of the RS-GW-JAX solver. Key Features: Differentiable Physics: Fully implemented in JAX for high-performance 5D warped geometry simulations. Goldberger-Wise Stabilization: Coupled Einstein-Scalar equations solved with high precision. Quantum Corrections: Built-in support for JT-Gravity and Schwarzian backreaction near the IR brane. Adaptive Integration: Custom RK45 solver optimized for XLA. Academic Standard: Includes self-auditing frameworks and effective Planck mass calculations. Verified against the Randall-Sundrum baseline with an accuracy of $10^{-14}$.
Publication details
- Journal / source
- Zenodo (CERN European Organization for Nuclear Research)
- Publication date
- 9 January 2026