Differentiable Optimization Python Packages

torchlie

A library for differentiable nonlinear optimization

torchopt

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

torchkin

A library for differentiable nonlinear optimization

theseus-ai

A library for differentiable nonlinear optimization

neuromancer

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

softtorch

A library for soft differentiable relaxations of common PyTorch functions.

dprox

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

softjax

A library for soft differentiable relaxations of common JAX functions.

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

Neural Modules with Adaptive Nonlinear Constraints and Efficient Regularization