MPAX is a hardware-accelerated, differentiable, batchable, and distributable solver for mathematical programming in JAX, designed to integrate with modern computational and deep learning workflows: ...
Abstract: Conjugate gradient (CG) methods are a class of important methods for solving linear equations and nonlinear optimization problems. In this paper, we propose a new stochastic CG algorithm ...
Abstract: This article aims to develop a general and designable distributed algorithm for solving linear algebraic equations (LAEs), which departs from the design framework based on orthogonal ...