jaxkern (pypi): Kernels in Jax.
Keywords: gaussian-processes|jax|machine-learning|bayesian