Hi,
You can find a Jacobian function in this gist: https://gist.github.com/apaszke/226abdf867c4e9d6698bd198f3b45fb7 33
Note that it will be very slow if you want to full jacobian for any medium/large image.