Conjugate Gradient Python Packages

pycgdescent

Python wrapper for CG_DESCENT.

torch-numopt

Classic numerical optimizaton methods for Pytorch neural networks

nonnegcg

(Python, R, C/C++) Minimize a function subject to non-negativity constraints on variables (conjugate gradient)

python-somax

Second-Order Methods in JAX

trnsolver

Linear solvers and eigendecomposition for AWS Trainium via NKI (cuSOLVER-equivalent) — Jacobi eigh, Cholesky/LU/QR factorizations, CG/GMRES iterative solvers, Newton-Schulz inverse square root.

gpcg

An implementation of a conjugate gradient algorithm (GPCG) for solving bound-constrained quadratic programs.