Metadata-Version: 2.4
Name: jax-snana
Version: 0.1.0
Summary: A JAX implementation of SNANA: supernova simulation, light-curve fitting and cosmology
Author-email: Matthew Grayling <mg2102@cam.ac.uk>
License-Expression: MIT
Project-URL: Repository, https://github.com/mattgrayling/jax-snana
Project-URL: Documentation, https://jax-snana.readthedocs.io
Project-URL: Issues, https://github.com/mattgrayling/jax-snana/issues
Keywords: supernova,snana,jax,cosmology,astronomy
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Science/Research
Classifier: Operating System :: POSIX :: Linux
Classifier: Operating System :: MacOS
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.12
Classifier: Topic :: Scientific/Engineering :: Astronomy
Classifier: Topic :: Scientific/Engineering :: Physics
Requires-Python: >=3.12
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: jax>=0.10.1
Requires-Dist: jaxlib>=0.10.1
Requires-Dist: numpy>=2.0
Requires-Dist: scipy>=1.13
Requires-Dist: astropy>=6.0
Requires-Dist: h5py>=3.10
Requires-Dist: pandas>=2.1
Requires-Dist: platformdirs>=4.0
Requires-Dist: isal>=1.2
Requires-Dist: pyarrow>=14.0
Requires-Dist: pyyaml>=6.0
Requires-Dist: numpyro>=0.22.0
Provides-Extra: docs
Requires-Dist: sphinx>=7.0; extra == "docs"
Requires-Dist: myst-parser>=2.0; extra == "docs"
Requires-Dist: sphinx-rtd-theme>=2.0; extra == "docs"
Provides-Extra: test
Requires-Dist: pytest; extra == "test"
Requires-Dist: pytest-xdist; extra == "test"
Requires-Dist: pytest-cov; extra == "test"
Requires-Dist: hypothesis; extra == "test"
Requires-Dist: sphinx>=7.0; extra == "test"
Requires-Dist: myst-parser>=2.0; extra == "test"
Requires-Dist: sphinx-rtd-theme>=2.0; extra == "test"
Provides-Extra: redback
Requires-Dist: redback-jax>=0.4.1; extra == "redback"
Dynamic: license-file

# jax-snana

GPU-accelerated supernova simulation, light curve fitting and cosmology in JAX

Developed and maintained by Matt Grayling (@mattgrayling).

## About
jax-snana is a JAX reimplementation of the supernova analysis framework SNANA, supporting survey
simulations, light curve fitting, BBC distance fitting and cosmological inference on CPU or GPU. It
also provides a compiled simulator for simulation-based inference, as discussed in Grayling 2026
(https://arxiv.org/abs/2610.19000).

## Installation and usage
jax-snana can be pip-installed via the command `pip install jax-snana`.

Detailed instructions on how to install and run jax-snana can be found here:
https://jax-snana.readthedocs.io

## Support
If you find something unclear, come across any bugs or would like a feature that is not yet
implemented, please raise a GitHub issue.

## Citing the code
If you use jax-snana, please cite Grayling 2026 (arXiv:2610.19000) as well as SNANA, Kessler et al.
2009 (PASP 121, 1028). If you use any of the models included within jax-snana, please also cite the
papers that present them, listed here: https://jax-snana.readthedocs.io/en/latest/citing.html
