I’d like to ask whether Keras offers a backend-agnostic equivalent of jax.jvp; if not, I can contribute an implementation.