Computing the Jacobian for a MLP (in most non-conv MLP)