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.

Optimistix

Optimistix is a JAX library for nonlinear solves: root finding, minimisation, fixed points, and least squares. Optimistix is very fast, and has a number of modularity features designed to speed up optimisation research and simplify advanced optimisation tasks.