tinygp
the tiniest of Gaussian Process libraries

tinygp is an extremely lightweight library for building Gaussian Process (GP) models in Python, built on top of jax. It has a nice interface, and it's pretty fast. Thanks to jax, tinygp supports things like GPU acceleration and automatic differentiation.

Check out the docs for more info: tinygp.readthedocs.io