GPJax
Gaussian processes in JAX.
AI summary
GP model library
Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation
- stars
- 467
- forks
- 54
- watching
- 4
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
Model repository
Provides pre-trained deep learning models for the Jax/Flax ecosystem.
Gaussian process library
A Julia package that provides a flexible nonparametric tool for modeling data using Gaussian Processes
dfm/tinygp297
GP model
A lightweight library for building Gaussian Process models in Python
Gaussian process toolbox
Software for modeling and prediction with multiple output Gaussian processes
Kalman smoother
This project develops an approximate inference framework for temporal Gaussian processes using Kalman filtering and smoothing methods.
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
google/paxml461
ML framework
A framework for configuring and running machine learning experiments on top of Jax.
Uncertainty model
Software implementing probabilistic models for uncertainty estimation in data analysis
GP toolkit
A Matlab toolbox providing implementations of Gaussian processes and other machine learning tools.
Graph library
A lightweight library for working with graph neural networks in jax.
Bayesian optimizer
A Python package for Bayesian optimization using the GPFlow library and TensorFlow.
GP tool
An open-source Python package for applying Gaussian processes to images and hyperspectral data for reconstruction and Bayesian optimization.
Bayesian sampler library
A Bayesian inference library designed to provide modular and customizable samplers for probabilistic programming
openai/pixel-cnn1.9K
Generative model
A generative model with tractable likelihood and easy sampling, allowing for efficient data generation.
JAX tutorial series
A repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning.
Featured in 1 awesome list
Each link jumps to the spot where the list mentions GPJax.
