luas-gp 0.1.0__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- luas_gp-0.1.0/LICENSE +21 -0
- luas_gp-0.1.0/PKG-INFO +64 -0
- luas_gp-0.1.0/README.md +15 -0
- luas_gp-0.1.0/pyproject.toml +68 -0
- luas_gp-0.1.0/setup.cfg +4 -0
- luas_gp-0.1.0/src/luas/GP.py +2097 -0
- luas_gp-0.1.0/src/luas/__init__.py +14 -0
- luas_gp-0.1.0/src/luas/convenience_fns.py +1035 -0
- luas_gp-0.1.0/src/luas/exoplanet/__init__.py +8 -0
- luas_gp-0.1.0/src/luas/exoplanet/transit_fns.py +405 -0
- luas_gp-0.1.0/src/luas/kernel_interface.py +59 -0
- luas_gp-0.1.0/src/luas/kernel_selector.py +296 -0
- luas_gp-0.1.0/src/luas/kernels/BlockKernel.py +621 -0
- luas_gp-0.1.0/src/luas/kernels/Conditional.py +174 -0
- luas_gp-0.1.0/src/luas/kernels/DoubleMultiLowRankKernel.py +535 -0
- luas_gp-0.1.0/src/luas/kernels/ExtendedLuasKernel.py +578 -0
- luas_gp-0.1.0/src/luas/kernels/GeneralKernel.py +354 -0
- luas_gp-0.1.0/src/luas/kernels/LuasKernel.py +904 -0
- luas_gp-0.1.0/src/luas/kernels/LuasKernelND.py +268 -0
- luas_gp-0.1.0/src/luas/kernels/LuasLasrachKernel.py +525 -0
- luas_gp-0.1.0/src/luas/kernels/LuasPlusMultiLowRankKernel.py +587 -0
- luas_gp-0.1.0/src/luas/kernels/MixingMatGeneral.py +320 -0
- luas_gp-0.1.0/src/luas/kernels/MixingMatQuasisep.py +336 -0
- luas_gp-0.1.0/src/luas/kernels/MultiLowRankKernel.py +539 -0
- luas_gp-0.1.0/src/luas/kernels/SeparableKernel.py +322 -0
- luas_gp-0.1.0/src/luas/kernels/WhiteNoiseKernel.py +230 -0
- luas_gp-0.1.0/src/luas/kernels/__init__.py +20 -0
- luas_gp-0.1.0/src/luas/kernels/base.py +434 -0
- luas_gp-0.1.0/src/luas/kernels/covtype.py +1956 -0
- luas_gp-0.1.0/src/luas/kernels/data/matern_SSMs.npz +0 -0
- luas_gp-0.1.0/src/luas/kernels/data/matern_SSMs_stable.npz +0 -0
- luas_gp-0.1.0/src/luas/kernels/data/sq_exp_modal_decomps.npz +0 -0
- luas_gp-0.1.0/src/luas/kernels/diagonal.py +55 -0
- luas_gp-0.1.0/src/luas/kernels/householder.py +431 -0
- luas_gp-0.1.0/src/luas/kernels/lowrank.py +240 -0
- luas_gp-0.1.0/src/luas/kernels/modal_decomp_calc.py +261 -0
- luas_gp-0.1.0/src/luas/kernels/quasisep.py +329 -0
- luas_gp-0.1.0/src/luas/kernels/tinygp_ext.py +939 -0
- luas_gp-0.1.0/src/luas/kronecker_fns.py +271 -0
- luas_gp-0.1.0/src/luas/luas_types.py +28 -0
- luas_gp-0.1.0/src/luas/numpyro_ext.py +75 -0
- luas_gp-0.1.0/src/luas/priors.py +141 -0
- luas_gp-0.1.0/src/luas/pymc_ext.py +191 -0
- luas_gp-0.1.0/src/luas/stable_gradients.py +412 -0
- luas_gp-0.1.0/src/luas_gp.egg-info/PKG-INFO +64 -0
- luas_gp-0.1.0/src/luas_gp.egg-info/SOURCES.txt +53 -0
- luas_gp-0.1.0/src/luas_gp.egg-info/dependency_links.txt +1 -0
- luas_gp-0.1.0/src/luas_gp.egg-info/requires.txt +28 -0
- luas_gp-0.1.0/src/luas_gp.egg-info/top_level.txt +1 -0
- luas_gp-0.1.0/tests/test_covtype_algebra.py +90 -0
- luas_gp-0.1.0/tests/test_covtype_decompositions.py +95 -0
- luas_gp-0.1.0/tests/test_exoplanet_vs_batman.py +107 -0
- luas_gp-0.1.0/tests/test_gp_logL.py +102 -0
- luas_gp-0.1.0/tests/test_regression.py +77 -0
- luas_gp-0.1.0/tests/test_stress_2d.py +397 -0
luas_gp-0.1.0/LICENSE
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
MIT License
|
|
2
|
+
|
|
3
|
+
Copyright (c) 2023 Trinity College Dublin
|
|
4
|
+
|
|
5
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
6
|
+
of this software and associated documentation files (the "Software"), to deal
|
|
7
|
+
in the Software without restriction, including without limitation the rights
|
|
8
|
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
9
|
+
copies of the Software, and to permit persons to whom the Software is
|
|
10
|
+
furnished to do so, subject to the following conditions:
|
|
11
|
+
|
|
12
|
+
The above copyright notice and this permission notice shall be included in all
|
|
13
|
+
copies or substantial portions of the Software.
|
|
14
|
+
|
|
15
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
16
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
17
|
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
18
|
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
19
|
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
20
|
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
21
|
+
SOFTWARE.
|
luas_gp-0.1.0/PKG-INFO
ADDED
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
|
+
Name: luas-gp
|
|
3
|
+
Version: 0.1.0
|
|
4
|
+
Summary: A library for implementing efficient 2D Gaussian processes
|
|
5
|
+
Author-email: Mark Fortune <markfortuneis@gmail.com>
|
|
6
|
+
License-Expression: MIT
|
|
7
|
+
Project-URL: Documentation, https://luas.readthedocs.io
|
|
8
|
+
Project-URL: Source, https://github.com/markfortune/luas
|
|
9
|
+
Project-URL: Issues, https://github.com/markfortune/luas/issues
|
|
10
|
+
Project-URL: Paper, https://arxiv.org/abs/2611.09198
|
|
11
|
+
Classifier: Development Status :: 4 - Beta
|
|
12
|
+
Classifier: Intended Audience :: Developers
|
|
13
|
+
Classifier: Intended Audience :: Science/Research
|
|
14
|
+
Classifier: Operating System :: OS Independent
|
|
15
|
+
Classifier: Programming Language :: Python
|
|
16
|
+
Classifier: Programming Language :: Python :: 3
|
|
17
|
+
Classifier: Programming Language :: Python :: 3.12
|
|
18
|
+
Classifier: Programming Language :: Python :: 3.13
|
|
19
|
+
Classifier: Programming Language :: Python :: 3.14
|
|
20
|
+
Requires-Python: >=3.12
|
|
21
|
+
Description-Content-Type: text/markdown
|
|
22
|
+
License-File: LICENSE
|
|
23
|
+
Requires-Dist: jax>=0.4.38
|
|
24
|
+
Requires-Dist: numpy>=1.26.4
|
|
25
|
+
Requires-Dist: scipy>=1.11.4
|
|
26
|
+
Requires-Dist: tinygp>=0.3.1
|
|
27
|
+
Requires-Dist: equinox
|
|
28
|
+
Requires-Dist: matplotlib
|
|
29
|
+
Requires-Dist: tqdm
|
|
30
|
+
Provides-Extra: test
|
|
31
|
+
Requires-Dist: pytest; extra == "test"
|
|
32
|
+
Requires-Dist: jaxoplanet>=0.1.0; extra == "test"
|
|
33
|
+
Requires-Dist: astropy; extra == "test"
|
|
34
|
+
Provides-Extra: exoplanet
|
|
35
|
+
Requires-Dist: jaxoplanet>=0.1.0; extra == "exoplanet"
|
|
36
|
+
Requires-Dist: astropy; extra == "exoplanet"
|
|
37
|
+
Requires-Dist: pymc>=6; extra == "exoplanet"
|
|
38
|
+
Requires-Dist: corner>=2.3; extra == "exoplanet"
|
|
39
|
+
Requires-Dist: numpyro; extra == "exoplanet"
|
|
40
|
+
Provides-Extra: docs
|
|
41
|
+
Requires-Dist: luas-gp[exoplanet]; extra == "docs"
|
|
42
|
+
Requires-Dist: sphinx-book-theme; extra == "docs"
|
|
43
|
+
Requires-Dist: myst-nb; extra == "docs"
|
|
44
|
+
Requires-Dist: numpyro; extra == "docs"
|
|
45
|
+
Requires-Dist: pymc>=6; extra == "docs"
|
|
46
|
+
Requires-Dist: corner>=2.3; extra == "docs"
|
|
47
|
+
Requires-Dist: arviz; extra == "docs"
|
|
48
|
+
Dynamic: license-file
|
|
49
|
+
|
|
50
|
+
# luas
|
|
51
|
+
|
|
52
|
+
luas is a library focused on implementing fast and flexible 2D Gaussian processes using Kronecker product algebra, which assumes the data is rectangular and without missing values, but doesn't assume uniform sampling. We have also implemented ``luas`` for 1D datasets (e.g. time-series) and also N-dimensional datasets provided the kernel inputs lie on a grid. ``luas`` 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/.
|
|
53
|
+
|
|
54
|
+
See [Fortune et al. 2026](https://arxiv.org/abs/2611.09198) which explains the GP optimisations implemented in `luas`, while [Fortune et al. 2024](https://ui.adsabs.harvard.edu/abs/2024A%26A...686A..89F/abstract) is the original paper for ``luas`` which outlines the use of this package to analyse transmission spectroscopy observations.
|
|
55
|
+
|
|
56
|
+
## Installation
|
|
57
|
+
|
|
58
|
+
See the installation page on readthedocs [here](https://luas.readthedocs.io/en/latest/installation_info.html).
|
|
59
|
+
|
|
60
|
+
## License & Citing
|
|
61
|
+
|
|
62
|
+
`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). See the [docs](https://luas.readthedocs.io/en/latest/index.html) for more info on citing.
|
|
63
|
+
|
|
64
|
+
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](https://github.com/markfortune/luas/issues) or [send an email](mailto:mark.fortune@mq.edu.au).
|
luas_gp-0.1.0/README.md
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
# luas
|
|
2
|
+
|
|
3
|
+
luas is a library focused on implementing fast and flexible 2D Gaussian processes using Kronecker product algebra, which assumes the data is rectangular and without missing values, but doesn't assume uniform sampling. We have also implemented ``luas`` for 1D datasets (e.g. time-series) and also N-dimensional datasets provided the kernel inputs lie on a grid. ``luas`` 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/.
|
|
4
|
+
|
|
5
|
+
See [Fortune et al. 2026](https://arxiv.org/abs/2611.09198) which explains the GP optimisations implemented in `luas`, while [Fortune et al. 2024](https://ui.adsabs.harvard.edu/abs/2024A%26A...686A..89F/abstract) is the original paper for ``luas`` which outlines the use of this package to analyse transmission spectroscopy observations.
|
|
6
|
+
|
|
7
|
+
## Installation
|
|
8
|
+
|
|
9
|
+
See the installation page on readthedocs [here](https://luas.readthedocs.io/en/latest/installation_info.html).
|
|
10
|
+
|
|
11
|
+
## License & Citing
|
|
12
|
+
|
|
13
|
+
`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). See the [docs](https://luas.readthedocs.io/en/latest/index.html) for more info on citing.
|
|
14
|
+
|
|
15
|
+
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](https://github.com/markfortune/luas/issues) or [send an email](mailto:mark.fortune@mq.edu.au).
|
|
@@ -0,0 +1,68 @@
|
|
|
1
|
+
[build-system]
|
|
2
|
+
requires = ["setuptools>=77"]
|
|
3
|
+
build-backend = "setuptools.build_meta"
|
|
4
|
+
|
|
5
|
+
[project]
|
|
6
|
+
name = "luas-gp"
|
|
7
|
+
version = "0.1.0"
|
|
8
|
+
description = "A library for implementing efficient 2D Gaussian processes"
|
|
9
|
+
authors = [{ name = "Mark Fortune", email = "markfortuneis@gmail.com" }]
|
|
10
|
+
readme = "README.md"
|
|
11
|
+
requires-python = ">=3.12" # PyMC version 6 requires 3.12
|
|
12
|
+
license = "MIT"
|
|
13
|
+
license-files = ["LICENSE"]
|
|
14
|
+
# Minimum versions tested with full pytest run (Python 3.12, jax 0.4.38, numpy 1.26.4, scipy 1.11.4).
|
|
15
|
+
# Install tinygp from GitHub: the PyPI release is missing fixes luas needs.
|
|
16
|
+
dependencies = [
|
|
17
|
+
"jax>=0.4.38",
|
|
18
|
+
"numpy>=1.26.4",
|
|
19
|
+
"scipy>=1.11.4",
|
|
20
|
+
"tinygp>=0.3.1",
|
|
21
|
+
"equinox",
|
|
22
|
+
"matplotlib",
|
|
23
|
+
"tqdm",
|
|
24
|
+
]
|
|
25
|
+
classifiers = [
|
|
26
|
+
"Development Status :: 4 - Beta",
|
|
27
|
+
"Intended Audience :: Developers",
|
|
28
|
+
"Intended Audience :: Science/Research",
|
|
29
|
+
"Operating System :: OS Independent",
|
|
30
|
+
"Programming Language :: Python",
|
|
31
|
+
"Programming Language :: Python :: 3",
|
|
32
|
+
"Programming Language :: Python :: 3.12",
|
|
33
|
+
"Programming Language :: Python :: 3.13",
|
|
34
|
+
"Programming Language :: Python :: 3.14",
|
|
35
|
+
]
|
|
36
|
+
|
|
37
|
+
[project.urls]
|
|
38
|
+
Documentation = "https://luas.readthedocs.io"
|
|
39
|
+
Source = "https://github.com/markfortune/luas"
|
|
40
|
+
Issues = "https://github.com/markfortune/luas/issues"
|
|
41
|
+
Paper = "https://arxiv.org/abs/2611.09198"
|
|
42
|
+
|
|
43
|
+
[tool.setuptools.packages.find]
|
|
44
|
+
where = ["src"]
|
|
45
|
+
exclude = ["luas.redux*", "luas.kernels.experimental"]
|
|
46
|
+
|
|
47
|
+
[tool.setuptools.package-data]
|
|
48
|
+
"luas.kernels" = ["data/*"]
|
|
49
|
+
|
|
50
|
+
[project.optional-dependencies]
|
|
51
|
+
test = ["pytest", "jaxoplanet>=0.1.0", "astropy"]
|
|
52
|
+
exoplanet = ["jaxoplanet>=0.1.0", "astropy", "pymc>=6", "corner>=2.3", "numpyro"]
|
|
53
|
+
docs = [
|
|
54
|
+
"luas-gp[exoplanet]",
|
|
55
|
+
"sphinx-book-theme",
|
|
56
|
+
"myst-nb",
|
|
57
|
+
"numpyro",
|
|
58
|
+
"pymc>=6",
|
|
59
|
+
"corner>=2.3",
|
|
60
|
+
"arviz",
|
|
61
|
+
]
|
|
62
|
+
|
|
63
|
+
[tool.pytest.ini_options]
|
|
64
|
+
testpaths = ["tests"]
|
|
65
|
+
markers = [
|
|
66
|
+
"slow: slower tests such as gradient checks (deselect with -m \"not slow\")",
|
|
67
|
+
"stress: stress tests of unusual but reasonable kernel usage (deselect with -m \"not stress\")",
|
|
68
|
+
]
|
luas_gp-0.1.0/setup.cfg
ADDED