ott.solvers.linear.implicit_differentiation.ImplicitDiff.gradient# ImplicitDiff.gradient(prob, f, g, lse_mode, gr)[source]# Apply VJP to recover gradient in reverse mode differentiation. Parameters prob (LinearProblem) – f (Array) – g (Array) – lse_mode (bool) – gr (Tuple[Array, Array]) – Return type LinearProblem