JAX-bandflux: differentiable supernovae SALT modelling for cosmological analysis on GPUs

Zenodo (CERN European Organization for Nuclear Research) 2026 Open access

Read full text
Authors: S. A. K. Leeney
DOI: 10.5281/zenodo.18404557
Cited by: 0

Abstract

JAX-bandflux is a JAX implementation of critical supernova modelling functionality for cosmological analysis. The codebase implements key components of the established library SNCosmo in a differentiable framework, offering efficient parallelisation and gradient-based optimisation capabilities through GPU acceleration.

Fields of Study & Subjects

Gamma-ray bursts and supernovae Cosmology and Gravitation Theories Neutrino Physics Research

Publication details

Journal / source
Zenodo (CERN European Organization for Nuclear Research)
Publication date
28 January 2026