GPJax
GP model library
Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation
Gaussian processes in JAX.
467 stars
4 watching
54 forks
Language: Python
last commit: about 2 months ago
Linked from 1 awesome list
bayesian-inferencegaussian-processesjaxmachine-learningprobabilistic-programming
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 matthias-wright/flaxmodels | Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 240 |
 stor-i/gaussianprocesses.jl | A Julia package that provides a flexible nonparametric tool for modeling data using Gaussian Processes | 307 |
 dfm/tinygp | A lightweight library for building Gaussian Process models in Python | 297 |
 sheffieldml/multigp | Software for modeling and prediction with multiple output Gaussian processes | 48 |
 aaltoml/kalman-jax | This project develops an approximate inference framework for temporal Gaussian processes using Kalman filtering and smoothing methods. | 97 |
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
 google/paxml | A framework for configuring and running machine learning experiments on top of Jax. | 461 |
| sheffieldml/deepgp | Software implementing probabilistic models for uncertainty estimation in data analysis | 91 |
| sheffieldml/gpmat | A Matlab toolbox providing implementations of Gaussian processes and other machine learning tools. | 135 |
 google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,380 |
 gpflow/gpflowopt | A Python package for Bayesian optimization using the GPFlow library and TensorFlow. | 270 |
 ziatdinovmax/gpim | An open-source Python package for applying Gaussian processes to images and hyperspectral data for reconstruction and Bayesian optimization. | 57 |
 blackjax-devs/blackjax | A Bayesian inference library designed to provide modular and customizable samplers for probabilistic programming | 858 |
 openai/pixel-cnn | A generative model with tractable likelihood and easy sampling, allowing for efficient data generation. | 1,921 |
 gordicaleksa/get-started-with-jax | A repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning. | 670 |