JVP for Continuous differential operators

ucl-bug / jaxdf

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

GNU Lesser General Public License v3.0

117 stars 7 forks source link

Closed astanziola closed 9 months ago

astanziola commented 1 year ago

Really there's no reason to use vjp instead of jvp. I'm tempted to call this a bug.

astanziola commented 9 months ago

Well, turns out that differential operators are already using jax.jacfwd so... This was a non issue 😄