JaxRK is a library for working with (vectors of) RKHS elements and RKHS operators using JAX for automatic differentiation. This library includes implementations of kernel transfer operators and conditional density operators.
$ pip install jaxrk
For some examples of what you can do with JaxRK, see examples/Quick_start.ipynb.
To help in developing JaxRK, clone the github repo and change to the cloned directory on the command line. Then
$ pip install -e .
$ pytest test
will install the package into your python path. Changes to files in the directory are reflected in the python package when loaded.