KernaxΒΆ

JAX-based kernel library for Gaussian Processes.