gpax (pypi): Gaussian processes in NumPyro and JAX
Keywords: active-learning|bayesian-inference|bayesian-optimization|deep-kernel-learning|gaussian-processes|hypothesis-learning|machine-learning