Luis A. Ortega

Sure, this is a function from the Laplace library that computes the Jacobian of a model w.r.t to the parameters given an input [source](https://github.com/AlexImmer/Laplace/blob/e94a000adff1c9aef7dcba34e23fd7034e5cd8b6/laplace/curvature/backpack.py#L19): def jacobians(self, x): """Compute Jacobians \\(\\nabla_{\\theta}...

Hi @fKunstner The framework is similar to what you pointed in your first question, `x` is a batch of `a` dimensional points and I am trying to get `n x...