jaxopt
jaxopt copied to clipboard
Added Pseudo-inverse preconditioner for EqQP.
This allows to precompute a preconditioner, and share it across multiple outer loops, where the inner loop is solving an Equality Constrained QP. This should provide speedups when the parameters of the inner loop QP don't change too much.
TODO: modify the implicit diff decorator so that the jvp also uses the same preconditioner, since the backward system shares the same linear operator with the forward.
CC @Algue-Rythme, this is pretty similar to your Jacobi preconditioner for OSQP.