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

A generalized inverse for matrices

R. Penrose · 1955

Earlier work this paper cites.

The differentiation of pseudo-inverses and nonlinear least squares problems whose variables separate

Gene H. Golub and V. Pereyra · 1973

Earlier work this paper cites.

Digital Signal Processing

Alan V. Oppenheim and Ronald W. Schafer · 1975

Earlier work this paper cites.

Matlab version: 9.13.0 (r2022b), 1984

The MathWorks Inc · 1984

Earlier work this paper cites.

Gmres: A generalized minimal residual algorithm for solving nonsymmetric linear systems

Youcef Saad and Martin H. Schultz · 1986

Earlier work this paper cites.

Bi-cgstab: A fast and smoothly converging variant of bi-cg for the solution of nonsymmetric linear systems

H. A. van der Vorst · 1992

Earlier work this paper cites.

Numerical Partial Differential Equations: Finite Difference Methods

J. W. Thomas · 1995

Earlier work this paper cites.

Matrix Computations

Gene H. Golub and Charles F. Van Loan · 1996

Earlier work this paper cites.

Numerical Linear Algebra

Lloyd N. Trefethen and David Bau · 1997

Earlier work this paper cites.

Generalized Inverses: Theory and Applications

Adi Ben-Israel and Thomas N. E. Greville · 2003

Earlier work this paper cites.

Numerical Optimization (Second Edition)

Jorge Nocedal and Stephen Wright · 2006

Cited alongside, same era.

Julia: A fresh approach to numerical computing

Jeff Bezanson, Alan Edelman, Stefan Karpinski, and Viral B Shah · 2017

Cited alongside, same era.

Jax: composable transformations of python+numpy programs, 2018

James Bradbury, Roy Frostig, Peter Hawkins, Matthew James, Johnson, Chris Leary, Dougal Maclaurin, George Necula, Adam Paszke, Jake VanderPlas, Skye Wanderman-Milne, and Qiao Zhang · 2018

Cited alongside, same era.

The deepmind jax ecosystem, 2020

Igor Babuschkin et al · 2020

Cited alongside, same era.

Highly accurate protein structure prediction with alphafold

John Jumper et al · 2021

Cited alongside, same era.

Brax - a differentiable physics engine for large scale rigid body simulation, 2021

C. Daniel Freeman, Erik Frey, Anton Raichuk, Sertan Girgin, Igor Mordatch, and Olivier Bachem · 2021

Optimal transport tools (ott): A jax toolbox for all things wasserstein

Original

Marco Cuturi, Laetitia Meng-Papaxanthos, Yingtao Tian, Charlotte Bunne, Geoff Davis, and Olivier Teboul · 2022

Later among the works it cites.

Learning to correct spectral methods for simulating turbulent flows

Original

Gideon Dresdner, Dmitrii Kochkov, Peter Norgaard, Leonardo Zepeda-Núñez, Jamie A. Smith, Michael P. Brenner, and Stephan Hoyer · 2022

Later among the works it cites.

JAX-fluids: A fully-differentiable high-order computational fluid dynamics solver for compressible two-phase flows

Deniz A. Bezgin, Aaron B. Buhendwa, and Nikolaus A. Adams · 2023

Closest in time.

Linearsolve.jl: High-performance unified linear solvers, 2021

Chris Rackauckas et al · 2023

Closest in time.

Levanter — legible, scalable, reproducible foundation models with jax

David Hall, Ivan Zhou, and Percy Liang · 2023