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.

52.9M 42K 8K

ortools

Google's Operations Research tools:

6.9M 13K 2K

optax

Optax is a gradient processing and optimization library for JAX.

4.2M 2K 324

cvxpy

A Python-embedded modeling language for convex optimization problems.

3.9M 6K 1K

clarabel

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

3.1M 559 45

optimum

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

1.7M 3K 639

torch-model-archiver

Serve, optimize and scale PyTorch models in production

1.3M 4K 886

pyomo

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

1.2M 2K 577

onnxslim

A Toolkit to Help Optimize Onnx Model

1.2M 476 37

scikit-optimize

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

856K 24 6

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.

850K 2K 441

qpsolvers

Quadratic programming solvers in Python with a unified API

710K 743 102

opt-einsum-fx

Einsum optimization using opt_einsum and PyTorch FX graph rewriting

691K 22 5

jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

564K 1K 71

nlopt

A project to package the NLOpt library to wheels

470K 33 18

mip

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

382K 594 106

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.

336K 3K 778