jaxrk (pypi): JaxRK is a library for working with (vectors of) RKHS elements and RKHS operators using JAX for automatic differentiation.
Keywords: Jax|RKHS|kernel