gpjax (pypi): Gaussian processes in JAX.
Keywords: gaussian-processes|jax|machine-learning|bayesian|bayesian-inference|probabilistic-programming