pith. sign in

arxiv: 2311.17283 · v1 · pith:2H5H6H4Unew · submitted 2023-11-28 · 💻 cs.MS

Lineax: unified linear solves and linear least-squares in JAX and Equinox

Jason Rader , Terry Lyons , Patrick Kidger This is my paper
classification 💻 cs.MS
keywords linearlineaxleast-squaressolvesequinoxoperatorsautodifferentiableavailable
0
0 comments X
read the original abstract

We introduce Lineax, a library bringing linear solves and linear least-squares to the JAX+Equinox scientific computing ecosystem. Lineax uses general linear operators, and unifies linear solves and least-squares into a single, autodifferentiable API. Solvers and operators are user-extensible, without requiring the user to implement any custom derivative rules to get differentiability. Lineax is available at https://github.com/google/lineax.

This paper has not been read by Pith yet.

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Forward citations

Cited by 2 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. JAX-AMG: A GPU-Accelerated Differentiable Sparse Linear Solver Library for JAX

    cs.MS 2026-06 unverdicted novelty 7.0

    JAX-AMG is a new library that exposes AmgX AMG and Krylov methods as JAX primitives supporting JIT, reverse-mode AD, batched solves, and distributed execution.

  2. Complex surface patterning in homo- and heteroepitaxial contexts: (simultaneous) step bunching and step meandering

    cond-mat.mtrl-sci 2026-04 unverdicted novelty 5.0

    A linked discrete-continuum model demonstrates that step bunching and meandering can coexist as common growth modes on vicinal surfaces, yielding diverse patterns beyond separate limiting cases.