Papers › GODMAX: Modeling gas thermodynamics and matter distribution using JAX
GODMAX: Modeling gas thermodynamics and matter distribution using JAX
Shivam Pandey, Jaime Salcido, Chun-Hao To, J. Colin Hill, Dhayaa Anbajagane, Eric J. Baxter, Ian G. McCarthy
The archive published only this paper's code-link row. Authors, date and abstract are from arXiv's metadata (CC0), read from the Kaggle arXiv metadata snapshot of 2026-09-12 where its title matched the archive's; the title is the archive's.
We introduce GODMAX (Gas thermODynamics and Matter distribution using jAX), a novel code designed to calculate correlations between the cosmological matter distribution and various gas thermodynamic quantities. Utilizing the extensive ANTILLES suite of 200 hydrodynamical simulations with a diverse range of baryonic feedback strengths, we jointly fit the 3D profiles of total matter distribution, electron density, and electron pressure across various halo masses and redshifts. By accommodating significant variations in gas profiles expected due to baryonic feedback, solving exact hydrostatic equilibrium equation and offering flexible modeling of non-thermal pressure support, GODMAX has the capability to jointly fit all these profiles within the measurement uncertainties. This advancement enables, for the first time, robust joint analyses of multiple cosmic probes, including the kinetic and thermal Sunyaev-Zel'dovich effect, weak lensing, and X-ray observations. Furthermore, the model accurately captures correlations between the total matter power suppression due to baryonic feedback and local average thermodynamic quantities, such as the baryon fraction and integrated tSZ effect, in high-mass halos, aligning with observations from hydrodynamical simulations. Looking ahead, we forecast the expected constraints on cosmological and baryonic parameters from upcoming weak lensing catalogs from the LSST and tSZ maps from the Simons Observatory. This analysis underscores the importance of cross-correlations between weak lensing and tSZ in enhancing parameter constraints by resolving major systematic uncertainties due to baryonic physics. The GODMAX code leverages the JAX library, resulting in a fully differentiable halo model with native GPU compilation support.
Code
Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.
Code Syntology ran Syntology
Not run by Syntology. Nothing on this page verifies that the listed code works.
Results from the paper archive 2025-07-28
No leaderboard rows for this paper in the archive.
Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections