A library designed for implementing fast and flexible 2D Gaussian processes. It has been written in jax and can be easily used in combination with inference libraries such as PyMC and NumPyro. Documentation and tutorials are located at https://luas.readthedocs.io/en/latest/.
See Fortune et al. (2024) which explains the main optimisation included in luas and outlines the use of this package to analyse transmission spectroscopy observations.
Install by cloning and navigating into the downloaded directory, followed by entering the command:
$ pip install .
luas is licensed under an MIT license, feel free to use. We hope by making this package freely available and open source it will make it easier for people to account for systematics correlated across two dimensions in data sets, in addition to being helpful for any other applications (e.g. interpolation).
If you are using luas then please cite our work Fortune et al. (2024) with the BibTeX provided below:
@ARTICLE{2024A&A...686A..89F,
author = {{Fortune}, Mark and {Gibson}, Neale P. and {Foreman-Mackey}, Daniel and {Evans-Soma}, Thomas M. and {Maguire}, Cathal and {Ramkumar}, Swaetha},
title = "{How do wavelength correlations affect transmission spectra? Application of a new fast and flexible 2D Gaussian process framework to transiting exoplanet spectroscopy}",
journal = {\aap},
keywords = {methods: data analysis, methods: statistical, techniques: spectroscopic, planets and satellites: atmospheres, stars: individual: WASP-31, Astrophysics - Earth and Planetary Astrophysics, Astrophysics - Instrumentation and Methods for Astrophysics},
year = 2024,
month = jun,
volume = {686},
eid = {A89},
pages = {A89},
doi = {10.1051/0004-6361/202347613},
archivePrefix = {arXiv},
eprint = {2402.15204},
primaryClass = {astro-ph.EP},
adsurl = {https://ui.adsabs.harvard.edu/abs/2024A&A...686A..89F},
adsnote = {Provided by the SAO/NASA Astrophysics Data System}
}
We hope to expand the functionality of luas over time and welcome any help to do so. Also, if you encounter any issues, have any requests or questions then feel free to raise an issue or send an email.