Software

Lineax

Lineax is a JAX library for linear solves and linear least squares. Lineax handles abstract operators and general pytrees with stable gradients, fast compilation, and fast algorithms.