Differentiable Optimization Python Packages

torchopt

TorchOpt is an efficient library for differentiable optimization built upon PyTorch.

torchlie

A library for differentiable nonlinear optimization

torchkin

A library for differentiable nonlinear optimization

theseus-ai

A library for differentiable nonlinear optimization

softtorch

A library for soft differentiable relaxations of common PyTorch functions.

neuromancer

Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.

softjax

A library for soft differentiable relaxations of common JAX functions.

agnimhd

This stability code analyzes and optimizes against global pressure and current driven instabilities in magnetic confinement fusion devices

silva-networks

SILVA Networks is a Python package for extended deep equilibrium layers, fixed-point solvers, implicit differentiation, structured operators, diagnostics, and reproducible research notebooks.

dprox

Official code repository for ∇-Prox: Differentiable Proximal Algorithm Modeling for Large-Scale Optimization (SIGGRAPH TOG 2023)

theseus-ai-nightly

A library for differentiable nonlinear optimization

deepc-hunt

[L4DC 2025] Automatic hyperparameter tuning for DeePC. Built by Michael Cummins at the Automatic Control Laboratory, ETH Zurich.

geosimilarity

Differentiable curve and surface similarity measures.

birm-nm-foo

Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.

theseus-opt

A library for differentiable nonlinear optimization