Gaussian Processes Python Packages

gpytorch

A highly efficient implementation of Gaussian Processes in PyTorch

gpflow

Gaussian processes in TensorFlow

pykrige

Kriging Toolkit for Python

egobox

Efficient global optimization toolbox in Rust: bayesian optimization, mixture of gaussian processes, sampling methods

arm-mango

Parallel Hyperparameter Tuning in Python

george

Fast and flexible Gaussian Process regression in Python

gpjax

Gaussian processes in JAX and Equinox.

baytune

A simple, extensible library for developing AutoML systems

celerite

Scalable 1D Gaussian Processes in C++, Python, and Julia

gpboost

Tree-Boosting, Gaussian Processes, and Mixed-Effects Models

detkit

Matrix Determinant Toolkit (memdet)

mellon

Non-parametric density inference for single-cell analysis.

resona

One matrix-free spectral primitive for an operator's whole spectral world — densities, moments, free convolution, pseudospectra, the Brown measure, dequantization diagnostics. Spans 18+ fields of math & physics. pip install resona

fvgp

A software package for flexible HPC GPs

gaussforge

High-performance Gaussian Splatting format conversion library

robustcov

Robust scatter geometry for anomaly detection, kernels, and ML workflows in Python.

glearn

High-performance python package for machine learning using Gaussian process.

pysot

Surrogate Optimization Toolbox for Python

ptgp

GPs in PyTensor

neural-tangents

Fast and Easy Infinite Neural Networks in Python

mogp-emulator

Package for fitting Gaussian Process Emulators to multiple output computer simulation results.

numpy-ml

Machine learning, in numpy

eztao

A Python Toolkit for AGN Time Series Analysis using CARMA models

gpjax-nightly

Gaussian processes in JAX and Equinox.