Skip to main content

Hierarchical Bayesian modelling of type Ia SNe

Project description

BayeSN

Hierarchical Bayesian SED modelling of type Ia supernova light curves

Developed and maintained by: Matt Grayling (@mattgrayling), Stephen Thorp (@stevet40), Gautham Narayan (@gnarayan), and Kaisey S. Mandel (@CambridgeAstroStat) on behalf of the BayeSN Team (@bayesn).

About

BayeSN is a probabilistic optical-NIR SED model for type Ia supernovae, allowing for hierarchical analysis of the population distribution of physical properties as well as cosmology-independent distance estimation for individual SNe. This repository contains an implementation of the BayeSN SED model built with numpyro and jax, with support for GPU acceleration, as discussed in Grayling+2024 (arXiv link <https://arxiv.org/abs/2401.08755>, accepted by MNRAS).

Installation and usage

BayeSN can be pip-installed via the command pip install bayesn.

Detailed instructions on how to install and run the BayeSN model can be found here: https://bayesn.readthedocs.io/en/latest/index.html

Support

Our goal with BayeSN is for the package to be clear and easy to use. If you find something unclear, come across any bugs or think of any functionality that would add a lot of value for you, please raise a Github issue.

Citing the code/model

If you utilise the BayeSN model in any way, please cite Mandel+2022 (MNRAS 510, 3, 3939-3966), and if you make use of this code please cite Grayling+2024 (submitted to MNRAS). In addition, if you use any of the pre-trained models included within BayeSN please cite the corresponding papers which present those models:

  • M20 model: Mandel et al. 2022 (MNRAS 510, 3, 3939-3966)
  • T21 model: Thorp et al. 2021 (MNRAS 508, 3, 4310-4331)
  • W22 model: Ward et al. 2023 (ApJ 956, 2, 111)

Things to note

Filters

BayeSN includes a set of built-in filters for convenience, as detailed in the documentation. However, you can also add your own filters through a yaml file to allow for new and updated filter responses to be used without needing a package update.

GPU Acceleration

This code has been designed with GPU acceleration in mind, and running on a GPU should yield a considerable (~100 times) increase in performance. However, it is important to note that GPUs will show the most benefit running large scale jobs. If you want to fit samples of 100s of SNe, the fit time per object will be considerably shorter than fitting just a handful. With only 1 object, you are likely better off running on CPU than GPU.

Redshift requirement

At the present time, BayeSN is designed for use with spectroscopic redshifts, which are kept fixed during inference. However, fitting with redshift as a free parameter based on a photometric redshift prior is feasible and planned in future.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

bayesn-0.4.0.tar.gz (3.4 MB view details)

Uploaded Source

Built Distribution

bayesn-0.4.0-py3-none-any.whl (3.4 MB view details)

Uploaded Python 3

File details

Details for the file bayesn-0.4.0.tar.gz.

File metadata

  • Download URL: bayesn-0.4.0.tar.gz
  • Upload date:
  • Size: 3.4 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.11.4

File hashes

Hashes for bayesn-0.4.0.tar.gz
Algorithm Hash digest
SHA256 c536639ca18d12bfb5d24a5757f86ce03084a43ec076ee684ca03da0ed382519
MD5 9a6a605d52c5b60b04317db7d9664d4c
BLAKE2b-256 8f722c428af91a97bda5135895dd4dc0d3fc2004096039b74dedeed8330c7523

See more details on using hashes here.

File details

Details for the file bayesn-0.4.0-py3-none-any.whl.

File metadata

  • Download URL: bayesn-0.4.0-py3-none-any.whl
  • Upload date:
  • Size: 3.4 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.11.4

File hashes

Hashes for bayesn-0.4.0-py3-none-any.whl
Algorithm Hash digest
SHA256 30dc373e4cd556b736e7d8eb4b1f181142fd5a41caa60dd7befc50a99cc35313
MD5 b0d1f017e49c5ff1b740204e2ee4e7fa
BLAKE2b-256 047aa249ff3f0b11f713155f27266399deba1893def7e7a9bdb3229968c1565a

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page