Optimization Python Packages

ray

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

56.1M 43K 8K

ortools

Google's Operations Research tools:

4.9M 13K 2K

optax

Optax is a gradient processing and optimization library for JAX.

4.6M 2K 325

cvxpy

A Python-embedded modeling language for convex optimization problems.

4M 6K 1K

clarabel

Clarabel.rs: Interior-point solver for convex conic optimisation problems in Rust.

3.2M 562 45

optimum

🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools

1.9M 3K 640

pyomo

An object-oriented algebraic modeling language in Python for structured optimization problems.

1.3M 2K 577

opt-einsum-fx

Einsum optimization using opt_einsum and PyTorch FX graph rewriting

1.2M 22 5

onnxslim

A Toolkit to Help Optimize Onnx Model

1.2M 481 40

torch-model-archiver

Serve, optimize and scale PyTorch models in production

963K 4K 886

casadi

CasADi is a symbolic framework for numeric optimization implementing automatic differentiation in forward and reverse modes on sparse matrix-valued computational graphs. It supports self-contained C-code generation and interfaces state-of-the-art codes such as SUNDIALS, IPOPT etc. It can be used from C++, Python or Matlab/Octave.

840K 2K 444

scikit-optimize

Sequential model-based optimization with a `scipy.optimize` interface

795K 24 6

qpsolvers

Quadratic programming solvers in Python with a unified API

725K 748 103

jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

648K 1K 71

nlopt

A project to package the NLOpt library to wheels

500K 33 18

mip

Python-MIP: collection of Python tools for the modeling and solution of Mixed-Integer Linear programs

371K 595 107

daqp

A dual active-set algorithm for convex quadratic programming

346K 106 19

pennylane

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

312K 3K 782