mojogp 0.26.6.0__tar.gz

mojogp-0.26.6.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 caspbian
+
+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.

mojogp-0.26.6.0/MANIFEST.in

@@ -0,0 +1,15 @@
+include README.md
+include LICENSE
+recursive-include mojogp *.py *.mojo
+recursive-exclude * __pycache__ *.py[cod]
+exclude mojogp_jit_engine.so
+prune archive
+prune benchmark_data_vast
+prune isolated_engine_loads
+prune max_runtime_libs
+prune max_sdk_bundle
+prune node_modules
+prune ref
+prune results
+prune tests
+prune tmp

mojogp-0.26.6.0/PKG-INFO

@@ -0,0 +1,222 @@
+Metadata-Version: 2.4
+Name: mojogp
+Version: 0.26.6.0
+Summary: GPU-accelerated Gaussian Process library using Mojo kernels
+Author: caspbian
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/caspbian/mojogp
+Project-URL: Documentation, https://caspbian.github.io/mojogp/
+Project-URL: Repository, https://github.com/caspbian/mojogp
+Project-URL: Issues, https://github.com/caspbian/mojogp/issues
+Keywords: gaussian-processes,mojo,gpu,machine-learning
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Science/Research
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: <3.12,>=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy
+Requires-Dist: mojo==0.25.7.0
+Requires-Dist: max==25.7.0
+Requires-Dist: sympy>=1.12
+Requires-Dist: tqdm
+Provides-Extra: test
+Requires-Dist: gpytorch; extra == "test"
+Requires-Dist: linear_operator; extra == "test"
+Requires-Dist: torch>=2.11.0; extra == "test"
+Requires-Dist: pytest>=7.0; extra == "test"
+Requires-Dist: scipy>=1.7.0; extra == "test"
+Requires-Dist: matplotlib>=3.5.0; extra == "test"
+Requires-Dist: scikit-learn; extra == "test"
+Requires-Dist: properscoring>=0.1; extra == "test"
+Requires-Dist: psutil>=5.8.0; extra == "test"
+Provides-Extra: docs
+Requires-Dist: griffe>=1.14.0; extra == "docs"
+Requires-Dist: mkdocs-material>=9.6.0; extra == "docs"
+Requires-Dist: mkdocstrings[python]>=0.30.0; extra == "docs"
+Provides-Extra: notebooks
+Requires-Dist: marimo>=0.23.1; extra == "notebooks"
+Requires-Dist: matplotlib; extra == "notebooks"
+Requires-Dist: scikit-learn; extra == "notebooks"
+Requires-Dist: tqdm; extra == "notebooks"
+Provides-Extra: sm75
+Requires-Dist: mojogp-cuda-sm75==0.26.6.0; extra == "sm75"
+Provides-Extra: sm-75
+Requires-Dist: mojogp-cuda-sm75==0.26.6.0; extra == "sm-75"
+Provides-Extra: sm80
+Requires-Dist: mojogp-cuda-sm80==0.26.6.0; extra == "sm80"
+Provides-Extra: sm-80
+Requires-Dist: mojogp-cuda-sm80==0.26.6.0; extra == "sm-80"
+Provides-Extra: sm86
+Requires-Dist: mojogp-cuda-sm86==0.26.6.0; extra == "sm86"
+Provides-Extra: sm-86
+Requires-Dist: mojogp-cuda-sm86==0.26.6.0; extra == "sm-86"
+Provides-Extra: sm89
+Requires-Dist: mojogp-cuda-sm89==0.26.6.0; extra == "sm89"
+Provides-Extra: sm-89
+Requires-Dist: mojogp-cuda-sm89==0.26.6.0; extra == "sm-89"
+Provides-Extra: sm90
+Requires-Dist: mojogp-cuda-sm90==0.26.6.0; extra == "sm90"
+Provides-Extra: sm-90
+Requires-Dist: mojogp-cuda-sm90==0.26.6.0; extra == "sm-90"
+Provides-Extra: sm100
+Requires-Dist: mojogp-cuda-sm100==0.26.6.0; extra == "sm100"
+Provides-Extra: sm-100
+Requires-Dist: mojogp-cuda-sm100==0.26.6.0; extra == "sm-100"
+Provides-Extra: sm120
+Requires-Dist: mojogp-cuda-sm120==0.26.6.0; extra == "sm120"
+Provides-Extra: sm-120
+Requires-Dist: mojogp-cuda-sm120==0.26.6.0; extra == "sm-120"
+Dynamic: license-file
+
+# MojoGP
+
+> **Active development:** MojoGP is pre-1.0 software. Expect sharp edges, API changes, incomplete routes.
+
+MojoGP is a Python-first exact Gaussian Process regression library backed by JIT-compiled Mojo GPU kernels. It exists to make exact GP training practical without depending on Torch: the runtime package depends on NumPy, SymPy, tqdm, and Mojo/MAX.
+
+## Why MojoGP
+
+1. No torch dependency: NumPy, SymPy, tqdm, and Mojo/MAX only.
+2. Materialized and matrix-free routes: dense materialized kernels and matrix-free kernel matvecs.
+3. JIT-compiled GP models: models JIT-compile and kernels cached before training/prediction.
+4. Multi-output GPs: ICM-style and LMC-style.
+5. Discrete/categorical kernels: initial support for mixed continuous-categorical kernels.
+6. Support for NVIDIA GPUs - support for AMD and Apple Silicon is on the roadmap.
+
+## Features
+
+Mixed means continuous plus categorical/discrete inputs.
+
+| Feature | Single-output continuous | Single-output mixed | Multi-output ICM continuous | Multi-output ICM mixed | Multi-output LMC continuous | Multi-output LMC mixed |
+|---|---|---|---|---|---|---|
+| Materialized training | alpha | experimental | experimental | experimental | alpha | experimental |
+| Matrix-free training | alpha | experimental | experimental | experimental | alpha | experimental |
+| Mean-only prediction | alpha | experimental | experimental | experimental | alpha | experimental |
+| Exact variance | alpha | experimental | experimental | experimental | alpha | experimental |
+| LOVE variance | alpha | experimental | experimental | experimental | experimental | experimental |
+| Heterogeneous latent kernels | n/a | n/a | n/a | n/a | alpha | experimental |
+| Active dimensions | alpha | experimental | experimental | experimental | alpha | experimental |
+| ARD lengthscales | alpha | experimental | experimental | in development | alpha | in development |
+| Additive kernel composites | alpha | experimental | experimental | experimental | alpha | in development |
+| Product kernel composites | alpha | experimental | experimental | experimental | alpha | experimental |
+| Save / load | alpha | experimental | experimental | experimental | alpha | experimental |
+| Learned homoskedastic noise | alpha | experimental | experimental | experimental | alpha | experimental |
+| Fixed observation noise | alpha | in development | alpha | in development | alpha | in development |
+| Learned heteroskedastic noise | alpha | in development | not started | not started | not started | not started |
+| Grouped noise | alpha | in development | alpha | in development | unsupported | unsupported |
+| Posterior sampling | alpha | experimental | experimental | experimental | alpha | experimental |
+
+## Examples
+
+See [`notebooks/examples/`](notebooks/examples/) for runnable examples covering:
+
+- single-output GPs
+- multi-output workflows
+- predictive uncertainty
+- categorical variables
+- observation-noise variants
+- posterior sampling
+- model persistence
+
+## Install
+
+PyPI packages are in development. For now, install MojoGP from source.
+
+Minimum runtime dependencies are Python 3.10 or 3.11, NumPy, SymPy, tqdm,
+Mojo, and MAX. The current build has been tested with Mojo 0.25.7.0 and MAX
+25.7.0.
+
+
+## Build From Source
+
+For now, build MojoGP from source. This workflow clones the source, installs the
+Python package, and builds one JIT engine locally for the GPU target you choose:
+
+```bash
+git clone https://github.com/caspbian/mojogp.git
+cd mojogp
+pip install -e .
+mojo build mojogp/kernels/jit/jit_engine_bindings.mojo \
+  --emit shared-lib \
+  -I mojogp/ \
+  -o mojogp_jit_engine.so \
+  --target-accelerator sm_89
+```
+
+Use the Mojo accelerator target that matches your GPU:
+
+
+| GPU family | Target |
+|---|---|
+| T4 / RTX 20-series / Quadro RTX | `sm_75` |
+| A100 / A30 | `sm_80` |
+| A40 / A10 / A16 / A2 / RTX 30-series / RTX A-series | `sm_86` |
+| L4 / L40 / L40S / RTX 40-series / RTX Ada | `sm_89` |
+| GH200 / H100 / H200 | `sm_90` |
+| B200 / GB200 | `sm_100` |
+| RTX PRO Blackwell / GeForce RTX 50-series | `sm_120` |
+
+
+## Hello World
+
+```python
+import numpy as np
+from mojogp import RBF, SingleOutputGP
+
+rng = np.random.default_rng(0)
+X = np.linspace(-3, 3, 2000, dtype=np.float32).reshape(-1, 1)
+y = (np.sin(2.0 * X[:, 0]) + 0.05 * rng.standard_normal(len(X))).astype(np.float32)
+
+gp = SingleOutputGP(RBF())
+gp.fit(X, y, max_iterations=50, method="matrix_free")
+
+X_test = np.linspace(-4, 4, 128, dtype=np.float32).reshape(-1, 1)
+mean, std = gp.predict(X_test, return_std=True, variance_method="love")
+```
+
+## References
+Bonilla, E.V., Chai, K. and Williams, C. (2007). Multi-task Gaussian Process Prediction. [online] Neural Information Processing Systems. Available at: https://papers.nips.cc/paper_files/paper/2007/hash/66368270ffd51418ec58bd793f2d9b1b-Abstract.html.
+
+Bruinsma, W.P., Perim, E., Tebbutt, W., Scott, H.J., Solin, A. and Turner, R.E. (2019). Scalable Exact Inference in Multi-Output Gaussian Processes. [online] arXiv.org. Available at: https://arxiv.org/abs/1911.06287 [Accessed 22 May 2026].
+
+Charlier, B., Feydy, J., Glaunès, J.A., Collin, F.-D. and Durif, G. (2020). Kernel Operations on the GPU, with Autodiff, without Memory Overflows. [online] arXiv.org. Available at: https://arxiv.org/abs/2004.11127 [Accessed 22 May 2026].
+
+Chen, T., Huber, C., Lin, E. and Zaid, H. (2026). Preconditioning without a preconditioner using randomized block Krylov subspace methods. ETNA - Electronic Transactions on Numerical Analysis, [online] 65, pp.63–92. doi:https://doi.org/10.1553/etna_vol65s63.
+
+Dong, K., Eriksson, D., Nickisch, H., Bindel, D. and Wilson, A.G. (2017). Scalable Log Determinants for Gaussian Process Kernel Learning. [online] arXiv.org. Available at: https://arxiv.org/abs/1711.03481 [Accessed 22 May 2026].
+
+Gardner, J.R., Pleiss, G., Bindel, D., Weinberger, K.Q. and Wilson, A.G. (2021). GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Acceleration. arXiv:1809.11165 [cs, stat]. [online] Available at: https://arxiv.org/abs/1809.11165.
+
+Godoy, W., Melnichenko, T., Valero-Lara, P., Elwasif, W., Fackler, P., Ferreira Da Silva, R., Teranishi, K. and Vetter, J. (2025). Mojo: MLIR-based Performance-Portable HPC Science Kernels on GPUs for the Python Ecosystem. Proceedings of the SC ’25 Workshops of the International Conference for High Performance Computing, Networking, Storage and Analysis, [online] pp.2114–2128. doi:https://doi.org/10.1145/3731599.3767573.
+
+Harbrecht, H., Peters, M. and Schneider, R. (2012). On the low-rank approximation by the pivoted Cholesky decomposition. Applied numerical mathematics, 62(4), pp.428–440. doi:https://doi.org/10.1016/j.apnum.2011.10.001.
+
+Perez, R.C., Veiga, D. and Garnier, J. (2025). A reproducible comparative study of categorical kernels for Gaussian process regression, with new clustering-based nested kernels. [online] arXiv.org. Available at: https://arxiv.org/abs/2510.01840 [Accessed 22 May 2026].
+
+Peter, Wu, H. and Wu, C.Y. (2008). Gaussian Process Models for Computer Experiments With Qualitative and Quantitative Factors. 50(3), pp.383–396. doi:https://doi.org/10.1198/004017008000000262.
+
+Pleiss, G., Gardner, J.R., Weinberger, K.Q. and Wilson, A.G. (2018). Constant-Time Predictive Distributions for Gaussian Processes. [online] arXiv.org. Available at: https://arxiv.org/abs/1803.06058 [Accessed 22 May 2026].
+
+Rakitsch, B., Lippert, C., Borgwardt, K. and Stegle, O. (2026). It is all in the noise: Efficient multi-task Gaussian process inference with structured residuals. Advances in Neural Information Processing Systems, [online] 26. Available at: https://proceedings.neurips.cc/paper/2013/hash/59c33016884a62116be975a9bb8257e3-Abstract.html [Accessed 22 May 2026].
+
+Rasmussen, C.E. and Williams, C.K.I. (2008). Gaussian processes for machine learning. Cambridge, Mass. Mit Press.
+
+Roustant, O., Padonou, E., Deville, Y., Clément, A., Perrin, G., Giorla, J. and Wynn, H. (2018). Group kernels for Gaussian process metamodels with categorical inputs. [online] arXiv.org. Available at: https://arxiv.org/abs/1802.02368 [Accessed 22 May 2026].
+
+Saves, P., Diouane, Y., Bartoli, N., Lefebvre, T. and Morlier, J. (2023). A mixed-categorical correlation kernel for Gaussian process. Neurocomputing, [online] 550, p.126472. doi:https://doi.org/10.1016/j.neucom.2023.126472.
+
+Shashanka Ubaru, Chen, J. and Saad, Y. (2017). Fast Estimation of $tr(f(A))$ via Stochastic Lanczos Quadrature. SIAM Journal on Matrix Analysis and Applications, 38(4), pp.1075–1099. doi:https://doi.org/10.1137/16m1104974.
+
+Wilson, A.G. and Nickisch, H. (2026). Kernel Interpolation for Scalable Structured Gaussian Processes (KISS-GP). [online] arXiv.org. Available at: https://arxiv.org/abs/1503.01057 [Accessed 22 May 2026].
+
+Wilson, J.T., Borovitskiy, V., Terenin, A., Mostowsky, P. and Deisenroth, M.P. (2020). Pathwise Conditioning of Gaussian Processes. [online] arXiv.org. Available at: https://arxiv.org/abs/2011.04026 [Accessed 22 May 2026].
+
+Zhou, Q., Peter Z.G. Qian and Zhou, S. (2011). A Simple Approach to Emulation for Computer Models With Qualitative and Quantitative Factors. Technometrics, 53(3), pp.266–273. doi:https://doi.org/10.1198/tech.2011.10025.
+
+## License
+
+MIT

mojogp-0.26.6.0/README.md

@@ -0,0 +1,148 @@
+# MojoGP
+
+> **Active development:** MojoGP is pre-1.0 software. Expect sharp edges, API changes, incomplete routes.
+
+MojoGP is a Python-first exact Gaussian Process regression library backed by JIT-compiled Mojo GPU kernels. It exists to make exact GP training practical without depending on Torch: the runtime package depends on NumPy, SymPy, tqdm, and Mojo/MAX.
+
+## Why MojoGP
+
+1. No torch dependency: NumPy, SymPy, tqdm, and Mojo/MAX only.
+2. Materialized and matrix-free routes: dense materialized kernels and matrix-free kernel matvecs.
+3. JIT-compiled GP models: models JIT-compile and kernels cached before training/prediction.
+4. Multi-output GPs: ICM-style and LMC-style.
+5. Discrete/categorical kernels: initial support for mixed continuous-categorical kernels.
+6. Support for NVIDIA GPUs - support for AMD and Apple Silicon is on the roadmap.
+
+## Features
+
+Mixed means continuous plus categorical/discrete inputs.
+
+| Feature | Single-output continuous | Single-output mixed | Multi-output ICM continuous | Multi-output ICM mixed | Multi-output LMC continuous | Multi-output LMC mixed |
+|---|---|---|---|---|---|---|
+| Materialized training | alpha | experimental | experimental | experimental | alpha | experimental |
+| Matrix-free training | alpha | experimental | experimental | experimental | alpha | experimental |
+| Mean-only prediction | alpha | experimental | experimental | experimental | alpha | experimental |
+| Exact variance | alpha | experimental | experimental | experimental | alpha | experimental |
+| LOVE variance | alpha | experimental | experimental | experimental | experimental | experimental |
+| Heterogeneous latent kernels | n/a | n/a | n/a | n/a | alpha | experimental |
+| Active dimensions | alpha | experimental | experimental | experimental | alpha | experimental |
+| ARD lengthscales | alpha | experimental | experimental | in development | alpha | in development |
+| Additive kernel composites | alpha | experimental | experimental | experimental | alpha | in development |
+| Product kernel composites | alpha | experimental | experimental | experimental | alpha | experimental |
+| Save / load | alpha | experimental | experimental | experimental | alpha | experimental |
+| Learned homoskedastic noise | alpha | experimental | experimental | experimental | alpha | experimental |
+| Fixed observation noise | alpha | in development | alpha | in development | alpha | in development |
+| Learned heteroskedastic noise | alpha | in development | not started | not started | not started | not started |
+| Grouped noise | alpha | in development | alpha | in development | unsupported | unsupported |
+| Posterior sampling | alpha | experimental | experimental | experimental | alpha | experimental |
+
+## Examples
+
+See [`notebooks/examples/`](notebooks/examples/) for runnable examples covering:
+
+- single-output GPs
+- multi-output workflows
+- predictive uncertainty
+- categorical variables
+- observation-noise variants
+- posterior sampling
+- model persistence
+
+## Install
+
+PyPI packages are in development. For now, install MojoGP from source.
+
+Minimum runtime dependencies are Python 3.10 or 3.11, NumPy, SymPy, tqdm,
+Mojo, and MAX. The current build has been tested with Mojo 0.25.7.0 and MAX
+25.7.0.
+
+
+## Build From Source
+
+For now, build MojoGP from source. This workflow clones the source, installs the
+Python package, and builds one JIT engine locally for the GPU target you choose:
+
+```bash
+git clone https://github.com/caspbian/mojogp.git
+cd mojogp
+pip install -e .
+mojo build mojogp/kernels/jit/jit_engine_bindings.mojo \
+  --emit shared-lib \
+  -I mojogp/ \
+  -o mojogp_jit_engine.so \
+  --target-accelerator sm_89
+```
+
+Use the Mojo accelerator target that matches your GPU:
+
+
+| GPU family | Target |
+|---|---|
+| T4 / RTX 20-series / Quadro RTX | `sm_75` |
+| A100 / A30 | `sm_80` |
+| A40 / A10 / A16 / A2 / RTX 30-series / RTX A-series | `sm_86` |
+| L4 / L40 / L40S / RTX 40-series / RTX Ada | `sm_89` |
+| GH200 / H100 / H200 | `sm_90` |
+| B200 / GB200 | `sm_100` |
+| RTX PRO Blackwell / GeForce RTX 50-series | `sm_120` |
+
+
+## Hello World
+
+```python
+import numpy as np
+from mojogp import RBF, SingleOutputGP
+
+rng = np.random.default_rng(0)
+X = np.linspace(-3, 3, 2000, dtype=np.float32).reshape(-1, 1)
+y = (np.sin(2.0 * X[:, 0]) + 0.05 * rng.standard_normal(len(X))).astype(np.float32)
+
+gp = SingleOutputGP(RBF())
+gp.fit(X, y, max_iterations=50, method="matrix_free")
+
+X_test = np.linspace(-4, 4, 128, dtype=np.float32).reshape(-1, 1)
+mean, std = gp.predict(X_test, return_std=True, variance_method="love")
+```
+
+## References
+Bonilla, E.V., Chai, K. and Williams, C. (2007). Multi-task Gaussian Process Prediction. [online] Neural Information Processing Systems. Available at: https://papers.nips.cc/paper_files/paper/2007/hash/66368270ffd51418ec58bd793f2d9b1b-Abstract.html.
+
+Bruinsma, W.P., Perim, E., Tebbutt, W., Scott, H.J., Solin, A. and Turner, R.E. (2019). Scalable Exact Inference in Multi-Output Gaussian Processes. [online] arXiv.org. Available at: https://arxiv.org/abs/1911.06287 [Accessed 22 May 2026].
+
+Charlier, B., Feydy, J., Glaunès, J.A., Collin, F.-D. and Durif, G. (2020). Kernel Operations on the GPU, with Autodiff, without Memory Overflows. [online] arXiv.org. Available at: https://arxiv.org/abs/2004.11127 [Accessed 22 May 2026].
+
+Chen, T., Huber, C., Lin, E. and Zaid, H. (2026). Preconditioning without a preconditioner using randomized block Krylov subspace methods. ETNA - Electronic Transactions on Numerical Analysis, [online] 65, pp.63–92. doi:https://doi.org/10.1553/etna_vol65s63.
+
+Dong, K., Eriksson, D., Nickisch, H., Bindel, D. and Wilson, A.G. (2017). Scalable Log Determinants for Gaussian Process Kernel Learning. [online] arXiv.org. Available at: https://arxiv.org/abs/1711.03481 [Accessed 22 May 2026].
+
+Gardner, J.R., Pleiss, G., Bindel, D., Weinberger, K.Q. and Wilson, A.G. (2021). GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Acceleration. arXiv:1809.11165 [cs, stat]. [online] Available at: https://arxiv.org/abs/1809.11165.
+
+Godoy, W., Melnichenko, T., Valero-Lara, P., Elwasif, W., Fackler, P., Ferreira Da Silva, R., Teranishi, K. and Vetter, J. (2025). Mojo: MLIR-based Performance-Portable HPC Science Kernels on GPUs for the Python Ecosystem. Proceedings of the SC ’25 Workshops of the International Conference for High Performance Computing, Networking, Storage and Analysis, [online] pp.2114–2128. doi:https://doi.org/10.1145/3731599.3767573.
+
+Harbrecht, H., Peters, M. and Schneider, R. (2012). On the low-rank approximation by the pivoted Cholesky decomposition. Applied numerical mathematics, 62(4), pp.428–440. doi:https://doi.org/10.1016/j.apnum.2011.10.001.
+
+Perez, R.C., Veiga, D. and Garnier, J. (2025). A reproducible comparative study of categorical kernels for Gaussian process regression, with new clustering-based nested kernels. [online] arXiv.org. Available at: https://arxiv.org/abs/2510.01840 [Accessed 22 May 2026].
+
+Peter, Wu, H. and Wu, C.Y. (2008). Gaussian Process Models for Computer Experiments With Qualitative and Quantitative Factors. 50(3), pp.383–396. doi:https://doi.org/10.1198/004017008000000262.
+
+Pleiss, G., Gardner, J.R., Weinberger, K.Q. and Wilson, A.G. (2018). Constant-Time Predictive Distributions for Gaussian Processes. [online] arXiv.org. Available at: https://arxiv.org/abs/1803.06058 [Accessed 22 May 2026].
+
+Rakitsch, B., Lippert, C., Borgwardt, K. and Stegle, O. (2026). It is all in the noise: Efficient multi-task Gaussian process inference with structured residuals. Advances in Neural Information Processing Systems, [online] 26. Available at: https://proceedings.neurips.cc/paper/2013/hash/59c33016884a62116be975a9bb8257e3-Abstract.html [Accessed 22 May 2026].
+
+Rasmussen, C.E. and Williams, C.K.I. (2008). Gaussian processes for machine learning. Cambridge, Mass. Mit Press.
+
+Roustant, O., Padonou, E., Deville, Y., Clément, A., Perrin, G., Giorla, J. and Wynn, H. (2018). Group kernels for Gaussian process metamodels with categorical inputs. [online] arXiv.org. Available at: https://arxiv.org/abs/1802.02368 [Accessed 22 May 2026].
+
+Saves, P., Diouane, Y., Bartoli, N., Lefebvre, T. and Morlier, J. (2023). A mixed-categorical correlation kernel for Gaussian process. Neurocomputing, [online] 550, p.126472. doi:https://doi.org/10.1016/j.neucom.2023.126472.
+
+Shashanka Ubaru, Chen, J. and Saad, Y. (2017). Fast Estimation of $tr(f(A))$ via Stochastic Lanczos Quadrature. SIAM Journal on Matrix Analysis and Applications, 38(4), pp.1075–1099. doi:https://doi.org/10.1137/16m1104974.
+
+Wilson, A.G. and Nickisch, H. (2026). Kernel Interpolation for Scalable Structured Gaussian Processes (KISS-GP). [online] arXiv.org. Available at: https://arxiv.org/abs/1503.01057 [Accessed 22 May 2026].
+
+Wilson, J.T., Borovitskiy, V., Terenin, A., Mostowsky, P. and Deisenroth, M.P. (2020). Pathwise Conditioning of Gaussian Processes. [online] arXiv.org. Available at: https://arxiv.org/abs/2011.04026 [Accessed 22 May 2026].
+
+Zhou, Q., Peter Z.G. Qian and Zhou, S. (2011). A Simple Approach to Emulation for Computer Models With Qualitative and Quantitative Factors. Technometrics, 53(3), pp.266–273. doi:https://doi.org/10.1198/tech.2011.10025.
+
+## License
+
+MIT

mojogp-0.26.6.0/mojogp/__init__.py

@@ -0,0 +1,162 @@
+"""MojoGP public Python API.
+
+The supported wrapper surface is centered on:
+
+1. `SingleOutputGP`
+2. `MultiOutputGP`
+3. `MultiOutputLMCGP`
+
+Quick example:
+    >>> import numpy as np
+    >>> from mojogp import SingleOutputGP as SoGP, RBF
+    >>> X = np.random.randn(5000, 2).astype(np.float32)
+    >>> y = (np.sin(X[:, 0]) + 0.1 * np.random.randn(5000)).astype(np.float32)
+    >>> gp = SoGP(RBF(ard=True))
+    >>> gp.fit(X, y, max_iterations=50, method="matrix_free")
+    >>> X_test = np.random.randn(100, 2).astype(np.float32)
+    >>> mean, std = gp.predict(X_test, return_std=True)
+
+Composite kernel example:
+    >>> from mojogp import SingleOutputGP as SoGP, RBF, Matern52
+    >>> kernel = RBF(active_dims=[0, 1]) + Matern52(active_dims=[1, 2])
+    >>> gp = SoGP(kernel)
+    >>> gp.fit(X, y, max_iterations=50, method="materialized")
+    >>> pred = gp.predict(X_test, variance_method="exact")
+
+Multi-output example:
+    >>> from mojogp import MultiOutputGP, MultiOutputLMCGP
+    >>> Y = np.random.randn(5000, 2).astype(np.float32)
+    >>> icm = MultiOutputGP(kernel="rbf")
+    >>> icm.fit(X, Y, max_iterations=30, method="matrix_free")
+    >>> mean, var = icm.predict(X_test, return_var=True)
+    >>> lmc = MultiOutputLMCGP(kernels=["rbf", "matern52"])
+    >>> lmc.fit(X, Y, max_iterations=30, method="materialized")
+
+Support boundaries worth knowing:
+
+1. pure categorical models are unsupported
+2. public `cholesky` posterior sampling is not part of the live API
+3. pathwise sampling is narrower than the full kernel surface
+4. `MultiOutputLMCGP(ard=True)` is available on the documented continuous LMC scope; mixed ARD applies only to continuous dimensions after categorical splitting
+5. fixed per-sample-per-task LMC observation noise `[n, T]` is available on the documented continuous LMC scope; mixed fixed noise is not public yet
+
+The generated feature/status matrix lives in `docs/FEATURE_MATRIX.md` and is
+backed by `mojogp.feature_support`.
+"""
+
+# Primary API - single-output GP with JIT-compiled composite kernels
+from mojogp.gp import (
+    SingleOutputGP,
+    TrainingResult,
+    PredictionResult,
+    fit_gp,
+)
+from mojogp._version import __version__
+from mojogp.feature_support import (
+    ExperimentalFeatureWarning,
+    FeatureStatus,
+    InDevelopmentFeatureWarning,
+    MojoGPFeatureWarning,
+)
+from mojogp import settings as settings
+from mojogp.settings import (
+    feature_warnings_suppressed,
+    get_feature_warnings_enabled,
+    get_progress_enabled,
+    progress_enabled,
+    set_feature_warnings_enabled,
+    set_progress_enabled,
+)
+from mojogp.progress import ProgressEvent, ProgressReporter
+
+# Kernel specification
+from mojogp.kernel import (
+    Kernel,
+    KernelNode,
+    KernelType,
+    make_ard_kernel,
+    compute_dim_permutation,
+    analyze_kernel_tree,
+    KernelTreeAnalysis,
+    CategoricalSpec,
+)
+from mojogp.kernel import (
+    RBF,
+    Matern12,
+    Matern32,
+    Matern52,
+    Periodic,
+    Linear,
+    RQ,
+    Polynomial,
+    # Categorical kernel aliases
+    GD,
+    CR,
+    EHH,
+    HH,
+    FE,
+)
+
+# Utilities
+from mojogp.utils import StandardScaler
+
+# Multi-output GP API
+from mojogp.multi_output_gp import (
+    MultiOutputGP,
+    MultiOutputLMCGP,
+    MultiOutputTrainingResult,
+    MultiOutputPredictionResult,
+    LMCTrainingResult,
+)
+
+__all__ = [
+    # Primary API
+    "SingleOutputGP",
+    "TrainingResult",
+    "PredictionResult",
+    "fit_gp",
+    # Multi-output GP
+    "MultiOutputGP",
+    "MultiOutputLMCGP",
+    "MultiOutputTrainingResult",
+    "MultiOutputPredictionResult",
+    "LMCTrainingResult",
+    # Kernel specification
+    "Kernel",
+    "KernelNode",
+    "KernelType",
+    "make_ard_kernel",
+    # Continuous kernel shortcuts
+    "RBF",
+    "Matern12",
+    "Matern32",
+    "Matern52",
+    "Periodic",
+    "Linear",
+    "RQ",
+    "Polynomial",
+    # Categorical kernel shortcuts
+    "GD",
+    "CR",
+    "EHH",
+    "HH",
+    "FE",
+    # Utilities
+    "StandardScaler",
+    # Version
+    "__version__",
+    # Feature support metadata/warnings
+    "FeatureStatus",
+    "MojoGPFeatureWarning",
+    "ExperimentalFeatureWarning",
+    "InDevelopmentFeatureWarning",
+    "settings",
+    "feature_warnings_suppressed",
+    "get_feature_warnings_enabled",
+    "get_progress_enabled",
+    "ProgressEvent",
+    "ProgressReporter",
+    "progress_enabled",
+    "set_feature_warnings_enabled",
+    "set_progress_enabled",
+]