jax-SNANA: A GPU-accelerated Supernova Framework for the SBI Era
astro-ph.IM, astro-ph.CO
Submitted: 2026-10-13
Updated: 2026-10-13
Comments: Submitted
Code: https://github.com/mattgrayling/jax-snana
License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
Sources
- Accelerating massive ensembles of ordinary differential equations
- Efficient and Modular Implicit Differentiation
- BlackJAX: Composable Bayesian inference in JAX
- Supernovae Unite: Combining Pantheon+ and DES-SN5YR
- LightCurveLynx: Forward Modeling of Time-Domain Surveys with Application to ZTF SN Ia DR2
- Simulation-based Inference with the Python Package sbijax
- BayeSN times Dovekie: Joint Photometric Cross-calibration and SED Modelling of Type Ia Supernovae
- SimSIMS: Simulation-based Supernova Ia Model Selection with thousands of latent variables
- Supernovae Unite: Host-Galaxy Mass Measurements of Type Ia Supernovae and Their Impact on Cosmology
- pmwd: A Differentiable Cosmological Particle-Mesh $N$-body Library
- Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro
- A Simulation Based Inference Approach to Modelling of Type Ia Supernova Populations
- skysurvey: a pure python package to simulate the transient sky
- A Reference Survey for Supernova Cosmology with the Nancy Grace Roman Space Telescope
- Banana Split: Improved Cosmological Constraints with Two Light-Curve-Shape and Color Populations Using Union3.1+UNITY1.8
- The LSST Dark Energy Science Collaboration (DESC) Science Requirements Document
- Hierarchical Simulation-Based Inference of Supernova Power Sources and their Physical Properties
- Amortized Bayesian Inference for Supernovae in the Era of the Vera Rubin Observatory Using Normalizing Flows
- Neural Posterior Estimation with Differentiable Simulators
Related papers
- A signal dedispersion algorithm for imaging-based transient searches
- EXPO: a quantum leap in fast, wide-band X-ray polarimetry for astrophysics
- Understanding cosmic explosions
- High-frequency water megamasers as precision probes of black holes, galaxies and cosmology: A case for a next-generation sub-mm/mm interferometer
- ALMA2040: Mapping the Spatially Resolved Dust and Star Formation Physics Driving High-z Galaxy Assembly
- Direct measures of black hole masses through cosmic time