Quadratic programming solvers in Python with a unified API
Quadratic programming solvers in Python with a unified API
This library provides a solve_qp function to solve convex quadratic programs:
$$ \begin{split} \begin{array}{ll} \underset{x}{\mbox{minimize}} & \frac{1}{2} x^T P x + q^T x \ \mbox{subject to} & G x \leq h \ & A x = b \ & lb \leq x \leq ub \end{array} \end{split} $$
Vector inequalities apply coordinate by coordinate. The function returns the primal solution $x^*$ found by the backend QP solver, or None in case of failure/unfeasible problem. All solvers require the problem to be convex, meaning the matrix $P$ should be positive semi-definite. Some solvers further require the problem to be strictly convex, meaning $P$ should be positive definite.
Dual multipliers: there is also a solve_problem function that returns not only the primal solution, but also its dual multipliers and all other relevant quantities computed by the backend solver.
To solve a quadratic program, build the matrices that define it and call solve_qp, selecting the backend QP solver via the solver keyword argument:
import numpy as np
from qpsolvers import solve_qp
M = np.array([[1.0, 2.0, 0.0], [-8.0, 3.0, 2.0], [0.0, 1.0, 1.0]])
P = M.T @ M # this is a positive definite matrix
q = np.array([3.0, 2.0, 3.0]) @ M
G = np.array([[1.0, 2.0, 1.0], [2.0, 0.0, 1.0], [-1.0, 2.0, -1.0]])
h = np.array([3.0, 2.0, -2.0])
A = np.array([1.0, 1.0, 1.0])
b = np.array([1.0])
x = solve_qp(P, q, G, h, A, b, solver="proxqp")
print(f"QP solution: {x = }")This example outputs the solution [0.30769231, -0.69230769, 1.38461538]. It is also possible to get dual multipliers at the solution, as shown in this example.
conda install -c conda-forge qpsolversTo install the library with open source QP solvers:
pip install qpsolvers[open_source_solvers]This one-size-fits-all installation may not work immediately on all systems (for instance if a solver tries to compile from source). If you run into any issue, check out the following variants:
pip install qpsolvers[wheels_only] will only install solvers with pre-compiled binaries,pip install qpsolvers[clarabel,daqp,proxqp,scs] (for instance) will install the listed set of QP solvers,pip install qpsolvers will only install the library itself.When imported, qpsolvers loads all the solvers it can find and lists them in qpsolvers.available_solvers.
| Solver | Keyword | Algorithm | API | License |
|---|---|---|---|---|
| Clarabel | clarabel |
Interior point | Sparse | Apache-2.0 |
| COPT | copt |
Interior point | Sparse | Commercial |
| CVXOPT | cvxopt |
Interior point | Dense | GPL-3.0 |
| DAQP | daqp |
Active set | Dense | MIT |
| ECOS | ecos |
Interior point | Sparse | GPL-3.0 |
| Gurobi | gurobi |
Interior point | Sparse | Commercial |
| HiGHS | highs |
Active set | Sparse | MIT |
| HPIPM | hpipm |
Interior point | Dense | BSD-2-Clause |
| jaxopt.OSQP | jaxopt_osqp |
Augmented Lagrangian | Dense | Apache-2.0 |
| KVXOPT | kvxopt |
Interior point | Dense & Sparse | GPL-3.0 |
| MOSEK | mosek |
Interior point | Sparse | Commercial |
| NPPro | nppro |
Active set | Dense | Commercial |
| OSQP | osqp |
Augmented Lagrangian | Sparse | Apache-2.0 |
| PDHCG | pdhcg |
Primal-dual hybrid gradient | Dense & Sparse | Apache-2.0 |
| PIQP | piqp |
Proximal interior point | Dense & Sparse | BSD-2-Clause |
| ProxQP | proxqp |
Augmented Lagrangian | Dense & Sparse | BSD-2-Clause |
| QPALM | qpalm |
Augmented Lagrangian | Sparse | LGPL-3.0 |
| qpmad | qpmad |
Active set | Dense | Apache-2.0 |
| QTQP | qtqp |
Interior point | Sparse | Apache-2.0 |
| qpax | qpax |
Interior point | Dense | MIT |
| qpOASES | qpoases |
Active set | Dense | LGPL-2.1 |
| qpSWIFT | qpswift |
Interior point | Sparse | GPL-3.0 |
| quadprog | quadprog |
Active set | Dense | GPL-2.0 |
| SCS | scs |
Augmented Lagrangian | Sparse | MIT |
| SIP | sip |
Barrier Augmented Lagrangian | Sparse | MIT |
Matrix arguments are NumPy arrays for dense solvers and SciPy Compressed Sparse Column (CSC) matrices for sparse ones.
QP solvers come with their strengths and weaknesses depending on the algorithmic choices they make. To help you find the ones most suited to your problems, you can check out the results from qpbenchmark, a benchmark for QP solvers in Python. The benchmark is divided into test sets, each test set representing a different distribution of quadratic programs with specific dimensions and structure (large sparse problems, optimal control problems, ...):
If you find this project useful, please consider giving it a :star: or citing it if your work is scientific:
…Don't forget to add yourself to the BibTeX above and to CITATION.cff if you contribute to this repository.
We welcome contributions! The first step is to install the library and use it. Report any bug in the issue tracker. If you're a developer looking to hack on open source, check out the contribution guidelines for suggestions.
No open issues yet, or sync has not completed.