JAX-bandflux：面向 GPU 的可微型超新星 SALT 模型
天体物理仪器与方法
2025-04-14 v1 宇宙学与河外天体物理
摘要
JAX-bandflux 是一个基于 JAX 实现的超新星建模工具包，用于宇宙学分析中的关键功能。该代码库在可微框架下实现了已建立库 SNCosmo 的关键组件，通过 GPU 加速提供高效的并行化与梯度优化能力。该软件包支持超新星光曲线测量的可微计算，为宇宙学分析推断 SALT 参数提供了可能性。
引用
@article{arxiv.2504.08081,
title = {JAX-bandflux: differentiable supernovae SALT modelling for cosmological analysis on GPUs},
author = {Samuel Alan Kossoff Leeney},
journal= {arXiv preprint arXiv:2504.08081},
year = {2025}
}
备注
Submitted to The Journal of Open Source Software on 10th April 2025