Closed astanziola closed 9 months ago
Really there's no reason to use vjp instead of jvp. I'm tempted to call this a bug.
Well, turns out that differential operators are already using jax.jacfwd so... This was a non issue 😄
jax.jacfwd
Really there's no reason to use vjp instead of jvp. I'm tempted to call this a bug.