Pith. sign in

REVIEW 5 cited by

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

Not yet reviewed by Pith; the record is open.

Re-run · record.json Download PDF Read on arXiv ↗

This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.

SPECIMEN: schema-true, not a live event

T0 review · schema-true

One-sentence machine reading of the paper's core claim.

pith:XXXXXXXX · record.json · timestamp

arxiv 2311.17283 v1 pith:2H5H6H4U 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
verification ladder T0 review T1 audit T2 compute T3 formal T4 reserved
0 comments
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.

discussion (0)

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

Forward citations

Cited by 5 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 conditional novelty 6.5

    A (2+1)D continuum Ginzburg–Landau step model and a double-well VicCA both produce coexisting step bunching and meandering with matching morphology diagrams across homo- and heteroepitaxial contexts.

  3. jaxdae: A JAX-native Differentiable Solver for Differential-Algebraic Equations in Coupled Multi-physics

    cs.MS 2026-07 conditional novelty 6.0

    jaxdae provides the first JAX-native differentiable DAE solver, using a frozen-grid BDF-2 replay adjoint for reverse-mode gradients and XLA-fused batched sweeps.

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

    cs.MS 2026-06 conditional novelty 6.0

    A new JAX library provides GPU-accelerated AMG-preconditioned sparse linear solves with reverse-mode automatic differentiation, JIT, and MPI support.

  5. 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.