nip.utils.maths.compute_neumann_ihvp