GPViz is an extension library to GPJax that facilitates fast and informative plotting of Gaussian processes and kernels.
All plots can be generated by simply calling the plot() command on the relevant Gaussian process object, a dataset, Jax PRNGKey, and a parameter set. For a full introduction, see the demo notebook here.
GPViz is available on PyPi and can be installed through
pip install gpviz