smolgp 0.0.1__tar.gz

smolgp-0.0.1/.gitignore

@@ -0,0 +1,16 @@
+*_version.py
+.coverage
+.nox
+.tox
+.coverage*
+/*.ipynb
+*.DAT
+docs/api/summary
+__pycache__
+*.egg-info
+uv.lock
+
+.python-version
+.DS_Store
+
+playground/

smolgp-0.0.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Ryan Rubenzahl
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

smolgp-0.0.1/PKG-INFO

@@ -0,0 +1,40 @@
+Metadata-Version: 2.4
+Name: smolgp
+Version: 0.0.1
+Summary: Gaussian Process State Space Models in Python/JAX
+Project-URL: Homepage, https://github.com/smolgp-dev/smolgp
+Project-URL: Issues, https://github.com/smolgp-dev/smolgp/issues
+Author-email: Ryan Rubenzahl <rrubenzahl@gmail.com>, Soichiro Hattori <soichiro.hattori@gmail.com>
+License: MIT
+License-File: LICENSE
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.10
+Requires-Dist: equinox>=0.13.1
+Requires-Dist: jax>=0.6.2
+Requires-Dist: tinygp>=0.3.0
+Description-Content-Type: text/markdown
+
+<p align="center">
+<img width="100" height="100" alt="smolgp-logo" src="https://github.com/user-attachments/assets/66c691c9-c4d3-4253-9587-82f50adda047"/><br>
+<strong>smolgp</strong><br>
+<i>State Space Models for O(Linear/Log) Gaussian Processes</i>
+</p>
+
+`smolgp` is a Python/JAX extension of the [`tinygp`](https://github.com/dfm/tinygp) package that implements
+1. A Kalman filter and RTS smoother as a `StateSpaceSolver` compatible with `tinygp`-like GP kernels.
+2. An `IntegratedStateSpaceSolver` that can handle integrated (and possibly overlapping) measurements from mutliple instruments (see Rubenzahl and Hattori et al. in prep)
+3. Parallelized versions of 1 (see [Särkkä and García-Fernández 2020](https://ieeexplore.ieee.org/document/9013038)) and 2 (see [Yaghoobi and Särkkä 2024](https://ieeexplore.ieee.org/abstract/document/10804629) and its [implementation](https://github.com/Fatemeh-Yaghoobi/Parallel-integrated-method?tab=readme-ov-file)) using `jax.lax.associative_scan`
+
+TODO:
+- benchmark plots from paper/showing full GP vs. QSM GP vs. SSM vs. parallel SSM
+- doc/example useage
+- tests
+
+Possible additions
+- define other kernels not in tinygp

smolgp-0.0.1/README.md

@@ -0,0 +1,18 @@
+<p align="center">
+<img width="100" height="100" alt="smolgp-logo" src="https://github.com/user-attachments/assets/66c691c9-c4d3-4253-9587-82f50adda047"/><br>
+<strong>smolgp</strong><br>
+<i>State Space Models for O(Linear/Log) Gaussian Processes</i>
+</p>
+
+`smolgp` is a Python/JAX extension of the [`tinygp`](https://github.com/dfm/tinygp) package that implements
+1. A Kalman filter and RTS smoother as a `StateSpaceSolver` compatible with `tinygp`-like GP kernels.
+2. An `IntegratedStateSpaceSolver` that can handle integrated (and possibly overlapping) measurements from mutliple instruments (see Rubenzahl and Hattori et al. in prep)
+3. Parallelized versions of 1 (see [Särkkä and García-Fernández 2020](https://ieeexplore.ieee.org/document/9013038)) and 2 (see [Yaghoobi and Särkkä 2024](https://ieeexplore.ieee.org/abstract/document/10804629) and its [implementation](https://github.com/Fatemeh-Yaghoobi/Parallel-integrated-method?tab=readme-ov-file)) using `jax.lax.associative_scan`
+
+TODO:
+- benchmark plots from paper/showing full GP vs. QSM GP vs. SSM vs. parallel SSM
+- doc/example useage
+- tests
+
+Possible additions
+- define other kernels not in tinygp

smolgp-0.0.1/pyproject.toml

@@ -0,0 +1,57 @@
+[project]
+name = "smolgp"
+version = "0.0.1"
+description = "Gaussian Process State Space Models in Python/JAX"
+authors = [{ name = "Ryan Rubenzahl", email = "rrubenzahl@gmail.com" },
+           { name = "Soichiro Hattori", email = "soichiro.hattori@gmail.com"}]
+readme = "README.md"
+requires-python = ">=3.10"
+license = { text = "MIT" }
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "Intended Audience :: Science/Research",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python",
+    "Programming Language :: Python :: 3",
+]
+dependencies = [
+    "equinox>=0.13.1",
+    "jax>=0.6.2",
+    "tinygp>=0.3.0",
+]
+
+[dependency-groups]
+dev = [
+    "astropy>=6.1.7",
+    "jupyter>=1.1.1",
+    "matplotlib>=3.10.6",
+]
+cuda12 = ["jax[cuda12]"]
+cuda13 = ["jax[cuda13]"]
+
+[tool.uv]
+conflicts = [
+    [
+    {group = "cuda12"}, 
+    {group = "cuda13"}
+    ]
+]
+
+[project.urls]
+Homepage = "https://github.com/smolgp-dev/smolgp"
+Issues = "https://github.com/smolgp-dev/smolgp/issues"
+
+[build-system]
+requires = ["hatchling >= 1.26"]
+build-backend = "hatchling.build"
+
+[tool.uv.sources]
+tinygp = { git = "https://github.com/dfm/tinygp" }
+
+[tool.ruff]
+ignore = ["E731", # lambda functions
+          "E741", # ambiguous variable names (e.g. I)
+          "E743", # ambiguous function names (e.g. I)
+         ]

smolgp-0.0.1/src/smolgp/__init__.py

@@ -0,0 +1,23 @@
+"""
+``smolgp`` is designed to be a drop-in extension of the `tinygp <https://github.com/dfm/tinygp>`_ 
+library for building Gaussian Process (GP) models in Python. It is also built 
+on top of `jax <https://github.com/google/jax>`_. ``smolgp`` uses state space
+representations of Gaussian Processes to implement linear-time (or up to logN with
+parallelization on GPU) solvers for GP regression and forecasting. It also implements 
+"integrated" kernels that can model time-averaged measurements, such as those from 
+long-exposure astronomical observations, which can also be solved in linear time and is
+compatible with the parallel methods.
+
+The primary way that you will use to interact with ``smolgp`` is by constructing
+"kernel" functions using the building blocks provided in the ``kernels``
+subpackage (see :ref:`api-kernels`), and then passing that to a
+:class:`GaussianProcess` object to do all the computations. Check out the
+:ref:`tutorials` for a more complete introduction.
+"""
+
+from smolgp import (
+    kernels as kernels,
+    solvers as solvers,
+)
+from smolgp.gp import GaussianProcess as GaussianProcess
+# from smolgp.smolgp_version import __version__ as __version__