Simple "Difference" Operator about jaxdf HOT 1 OPEN

I might have misunderstood, but basically what you are looking for is a jacobian operator?

That's in the following sense (I'll write in in 2d for simplicity).
Given a field $u(r) = (u_x(r), u_y(r))$ with $r \in \mathbb R^2$, you are looking for

$$ \nabla u = \begin{pmatrix} \frac{\partial u_x}{\partial x} & \frac{\partial u_x}{\partial y} \\ \frac{\partial u_y}{\partial x} & \frac{\partial u_y}{\partial y} \end{pmatrix} $$

If that's the case, we could add a jacobian operator, which is probably useful in general. Here's how it looks for finite differences: https://colab.research.google.com/notebooks/intro.ipynb

from jaxdf.