GitHub repository: JaxGaussianProcesses/GPJax
Primary language: Python
Description: Gaussian processes in JAX and Flax.