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.

nemo-relay

Multi-language agent runtime and library for execution scope management, lifecycle events, and middleware on tool and LLM calls.

ortools

Google's Operations Research tools:

cvxpy

A Python-embedded modeling language for convex optimization problems.

clarabel

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

optax

Optax is a gradient processing and optimization library for JAX.

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, Matlab/Octave, Julia or Javascript

nlopt

A project to package the NLOpt library to wheels

optimum

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

pyomo

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

qpsolvers

Quadratic programming solvers in Python with a unified API

bayesian-optimization

A Python implementation of global optimization with gaussian processes.

onnxslim

A Toolkit to Help Optimize Onnx Model

daqp

A dual active-set algorithm for convex quadratic programming

scikit-optimize

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

jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

opt-einsum-fx

Einsum optimization using opt_einsum and PyTorch FX graph rewriting

pandapower

Convenient Power System Modelling and Analysis based on PYPOWER and pandas

orloge

log parser for MIP and LP solvers

aerosandbox

Aircraft design optimization made fast through computational graph transformations (e.g., automatic differentiation). Composable analysis tools for aerodynamics, propulsion, structures, trajectory design, and much more.

optimistix

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

neuralfoil

NeuralFoil is a practical airfoil aerodynamics analysis tool using physics-informed machine learning, exposed to end-users in pure Python/NumPy.

cvxpy-base

A Python-embedded modeling language for convex optimization problems.

mip

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