jvp

fun jvp(x: DTensor, v: DTensor, f: (DTensor) -> DTensor): DTensor