patel-zeel/gpax

A JAX Gaussian process library for research purposes.

Jupyter Notebook