How to find jacobian of learnt neural network model

I modelled a neural network with a vector as input and a vector as output (multi input multi output NN model). Now I want to find jacobian of the learnt function as:

Would anyone please guide me how to do this in Julia (Flux)?

That’s not a gradient, that’s the Jacobian of f. See Zygote.jacobian.